今天遇到一个问题,就是寻找符合条件的元素所在的位置,主要使用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.argwhere,nonzero)插图数组中寻找符合条件元素的位置(np.argwhere,nonzero)插图(1)

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

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

方法2

a = np.argwhere(e>15)
print(a)
本站无任何商业行为
个人在线分享 » 数组中寻找符合条件元素的位置(np.argwhere,nonzero)
E-->