np.where()

numpy中能够返回符合某一条件的下标函数：np.where()，
—— np.where()[0] 表示行的索引；
—— np.where()[1] 则表示列的索引；

np.where()用于三目运算：
例如：如果A%2==0成立，则执行A+1，否则执行A-1

>>> A  
>>> array([1, 7, 4, 9, 2, 3, 6, 0, 8, 5])  
>>> B = np.where(A%2 == 0, A+1, A-1)   # 偶＋1，奇-1  
>>> B  
>>> array([0, 6, 5, 8, 3, 2, 7, 1, 9, 4])
>>>

numpy的函数

原文：https://www.cnblogs.com/yxl-2018/p/12416408.html