numpy.where用于子集而不是范围

假设来自数组:

[1,2,3,4]

我想找到元素24的索引。我可以使用:

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)...那样一一编写但与给定数组进行比较的情况下做到这一点?

yangleilgone 回答:numpy.where用于子集而不是范围

暂时没有好的解决方案,如果你有好的解决方案,请发邮件至:iooj@foxmail.com
本文链接:https://www.f2er.com/3158893.html

大家都在问