数组中寻找符合条件元素的位置(np.argwhere,nonzero)

本文主要是介绍数组中寻找符合条件元素的位置(np.argwhere,nonzero)，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

今天遇到一个问题，就是寻找符合条件的元素所在的位置，主要使用np.argwhere和nonzero函数

比如给我一个二维数组，我想知道其中元素大于15的位置

方法1

import numpy as np 
ex=np.arange(30)
e=np.reshape(ex,[3,10])
print(e)
print(e>15)

结果如下

np.transpose((e>15).nonzero())

方法2

a = np.argwhere(e>15)
print(a)

这篇关于数组中寻找符合条件元素的位置(np.argwhere,nonzero)的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---