[運算] 陣列中取多個特定值不用迴圈

看板MATLAB作者 (魯到入味)時間2年前 (), 2年前編輯推噓5(504)
留言9則, 4人參與, 2年前最新討論串1/1
大家好,小弟目前有兩個陣列a與b如下: A=[1 1 2 3 4 5 6 7 7 9]; B=[2 5 7]; 我想找出A中,與B中個別數字相同之元素,並輸出一邏輯陣列如下 [0 0 1 0 0 1 0 1 1 0] 該如何不使用for迴圈進行計算? 因為在實務上,A為一張最大可能達8kx8k的標籤(labeled)影像,而B為欲留下區域的標籤編號,可能也有數千個,目前卡在運送效率這裡。 謝謝大家 ----- Sent from JPTT on my Sony G8441. -- ※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 101.11.34.91 (臺灣) ※ 文章網址: https://www.ptt.cc/bbs/MATLAB/M.1561533820.A.BC8.html

06/26 15:34, 2年前 , 1F
sum(A==B'); 不確定有沒有其他內建指令可以用
06/26 15:34, 1F

06/26 15:39, 2年前 , 2F
感謝樓上大大!
06/26 15:39, 2F

06/26 19:25, 2年前 , 3F
應該是any
06/26 19:25, 3F

06/27 00:28, 2年前 , 4F
ismember()
06/27 00:28, 4F

06/27 00:33, 2年前 , 5F
ismember 可能比較好,或許會做 unique 以及最佳化搜尋。用
06/27 00:33, 5F

06/27 00:33, 2年前 , 6F
gpuarray 不知是否能加速,你試試。
06/27 00:33, 6F

06/27 13:13, 2年前 , 7F
ismember的確快得多
06/27 13:13, 7F

06/27 13:14, 2年前 , 8F
一開始的labeled image是從bwlabel來的嗎?
06/27 13:14, 8F

06/27 13:15, 2年前 , 9F
改用bwconncomp,其結果會比較方便處理
06/27 13:15, 9F
對的,我的二值影像是用bwlabel上標籤,現階段應該不會更動這段。 原先使用sin的方法,然後使用unique剔除重複值, 但是因為實際處理A與B累加過程的陣列過大,所以還是加了一個迴圈, 而運算速度從1.4ksec降到不到4sec,然後使用sppmg建議,僅需不到1.5sec, 速度提升三個數量級,原先整個程序最耗時的反而變得最不耗時。 謝謝大家 ※ 編輯: fxxkjoe1231 (1.169.143.221 臺灣), 06/27/2019 21:43:29
文章代碼(AID): #1T4nryl8 (MATLAB)
文章代碼(AID): #1T4nryl8 (MATLAB)