numpy中能够返回符合某一条件的下标函数:np.where(),只不过np.where()并不接受list类型的参数。
—— 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])
网友评论