今天遇到一个问题,就是寻找符合条件的元素所在的位置,主要使用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函数
比如给我一个二维数组,我想知道其中元素大于15的位置
import numpy as np
ex=np.arange(30)
e=np.reshape(ex,[3,10])
print(e)
print(e>15)
结果如下
np.transpose((e>15).nonzero())
a = np.argwhere(e>15)
print(a)