numpy函数 where使用详解，以及与 argwhere的 区别

import numpy as np

n2=np.array([[1, 2, 3],
                      [4, 5, 6],
                      [7, 8, 9]])

pos=np.where(n2>5)   #n2 找出所有大于5的元素的位置

n2[pos]                         #就是6，7  ，8， 9

#pos内容是([1,2,2,2],[3,0,1,2])  =（第一维度坐标列表，第二维度坐标列表）

#这个方法速度比其他的一般快

------------------------argwhere---------------------

如果你真想得到(1,3),(2,0),(2,1),(2,2)这样的数据就用np.argwhere

pos2=np.argwhere(n1>5)

print(pos2)