np.where用法

np.where用法总结
知乎大佬讲的更清楚
1.xx = np.array(xx)
np.where(condition, x, y) 相当于if xx的每一项是否满足条件 如果满足对应项变为x,否则变为y
2.输出条件中真值的坐标

3.输出数组中等于特定数值的索引

    arr = np.array([[1, 2], [3, 2], [9, 3], [7, 3]])
    print(arr)
    x = np.where(arr==3)
    print(x)

结果：
[[1 2]
[3 2]
[9 3]
[7 3]]
(array([1, 2, 3], dtype=int64), array([0, 1, 1], dtype=int64))
其中：array([1, 2, 3],表示行，array([0, 1, 1],表示列