假设来自数组:
[1,2,3,4]
我想找到元素2
和4
的索引。我可以使用:
arr = np.array([1,4])
np.where((arr == 2) | (arr == 4))
因此,我想将数组arr
与另一个给定的数组np.array([2,4])
进行比较并返回索引。如果我比较的数组不仅是2,4
,而是很多值,该怎么办?例如,假设我有一个元素为1,...,100
的改组数组,并且我想找到[1,4,15,27,88,12,...]
的索引。如何在不像(arr==1)| (arr==4)...
那样一一编写但与给定数组进行比较的情况下做到这一点?