python:numpy.where用法

np.where有两种用法：

1.np.where(condition, x, y):满足条件(condition)，输出x，不满足输出y。

2.np.where(condition)

只有条件 (condition)，没有x和y，则输出满足条件的元素的坐标。

这里的坐标以tuple的形式给出，通常原数组有多少维，输出的tuple中就包含几个数组，分别对应符合条件元素的各维坐标。

3d== (array([0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2]), 
      array([2, 2, 2, 0, 0, 0, 1, 1, 1, 2, 2, 2, 0, 0, 0, 1, 1, 1, 2, 2, 2]), 
      array([0, 1, 2, 0, 1, 2, 0, 1, 2, 0, 1, 2, 0, 1, 2, 0, 1, 2, 0, 1, 2]))

对于第一个返回的结果，我们可以看到，它返回的是一个元组tuple，

元组的第一个元素是一个array，它的值来源于满足条件的元素的页索引，三个0页的，9个1页的，9个2页的；

元祖的第二个元素是一个array，它的值来源于满足条件的元素的行索引。

元祖的第三个元素是一个array，它的值来源于满足条件的元素的列索引。

参考:

1.https://www.cnblogs.com/massquantity/p/8908859.html

2.https://blog.csdn.net/qq_27825451/article/details/82838402