我有一個1d陣列match和2d陣列data。我可以像這樣搜索一個或幾個元素：
data[data[:,0] == match[0] ]
data[data[:,0] == match[0] & data[:,0] == match[1] ]
但是你如何搜索所有的人 :
data[ data[:,0] == match ] 。
我想在numba函式里面我可以使用in1d()! :
mask = np.zeros(len(ar1), dtype=bool)
for a in ar2:
mask |= (ar1 == a)
return mask
uj5u.com熱心網友回復：
你可以使用numpy.isin:
nump.isin(data[:, 0], match)
示例：
data = np. array([[1, 4], [5, 2], [2, 4]])
match = np.array([2, 4] )
np.isin(data[:,0] , match)
# array（[False, False, True]）
data[np.isin(data[:, 0], match)] 。
# array([[2, 4]])
轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/326864.html
標籤：
上一篇：如何計算矩陣和矢量的距離？