假設我們有一個陣列A和一組陣列B。查找A每個陣列中存在多少項的快速方法是B什么?例如,如果我有:
A = np.array([2, 5, 10])
B = np.array([(3, 5, 10), (1, 2, 5), (4, 6, 9)])
所以我應該得到C = [1, 2, 0].
uj5u.com熱心網友回復:
您還可以使用廣播:
(B==A[:,None,None]).sum((0,2))
輸出:
array([2, 2, 0])
uj5u.com熱心網友回復:
np.count_nonzero(np.isin(B, A), axis=1)可能是實作您尋找的結果的一種方式。或者,以更優雅的方式 - 根據@mozway 的建議 - np.isin(B, A).sum(axis=1)。
轉載請註明出處,本文鏈接:https://www.uj5u.com/yidong/394725.html
上一篇:在方法中回傳類屬性后覆寫類屬性
