我有以下变量:
将numpy作为np导入
gens = np.array([2, 1, 2, 1, 0, 1, 2, 1, 2])
p = [0,1]
我想返回与p
的每个元素匹配的gens
的条目
因此,理想情况下,我希望它返回:
result = [[4],[2,3,5,7],[0,2,6,8]]
#[[where matched 0], [where matched 1], [the rest]]
---
到目前为止,我的尝试只适用于一个变量:
indx = gens.argsort()
res = np.searchsorted(gens[indx], [0])
gens[res] #gives 4, which is the position of 0
但我试着
indx = gens.argsort()
res = np.searchsorted(gens[indx], [1])
gens[res] #gives 1, which is the position of the first 1.
所以:
你可以使用np.where
或np.in1d和np.nonzero
相关问题 更多 >
编程相关推荐