python numpy的where函数

版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。

本文链接： https://blog.csdn.net/qq_43523725/article/details/102739787

arr = np.array(
    [[1,2,3],
     [2,3,4]]
)
print(arr.shape)  #(2,3)
t_index = np.where(arr>2)
print(t_index)
#(array([0, 1, 1], dtype=int64), array([2, 1, 2], dtype=int64))
#表示坐标为（0，2）（1，1）（1，2），即3，3，4三个位置的值大于2