np.where()用法解析

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

import numpy as np
aa = np.arange(10)
np.where(aa > 5,1,-1)

array([-1, -1, -1, -1, -1, -1, 1, 1, 1, 1])

2. np.where(condition)
   只有条件 (condition)，没有x和y，则输出满足条件 (即非0) 元素的坐标 (等价于numpy.nonzero)。这里的坐标以tuple的形式给出，通常原数组有多少维，输出的tuple中就包含几个数组，分别对应符合条件元素的各维坐标。

a = np.array([2,4,6,8,10])
b = np.where(a>5)
print (b)

(array([2, 3, 4], dtype=int64),)

参考链接：https://www.cnblogs.com/massquantity/p/8908859.html