python numpy库中where用法

python整段注释Ctrl+/

import numpy as np
np_where = np.where([[False,True],[True,True]],[[1,3],[2,4]],[[4,2],[6,5]])
print(np_where)  #[[4 3][2 4]]

where用法一 ：

np.where(x) x为数组，表示返回其中非0元素的下标

x = np.array([1,2,3]) #(array([0, 1, 2], dtype=int64),)
y = np.array([[0,0,3],[4,5,0]]) #(array([0, 1, 1], dtype=int64), array([2, 0, 1], dtype=int64))
print(np.where(x))
print(np.where(y))

where用法二 ：

给where()函数传入一个判断条件，返回TRUE的位置，若见该位置再传入数组中则可返回符合条件的元素

x = np.array([1,2,3]) 
y = np.array([[0,0,3],[4,5,]])
print(np.where(x%2 == 1)) #(array([0, 2], dtype=int64),)
print(x[np.where(x%2 == 1)]) #[1 3]
print(np.where(y>1)) #(array([0, 1, 1], dtype=int64), array([2, 0, 1], dtype=int64))
print(y[np.where(y>1)]) #[3 4 5]

where用法三：

where(condition, x=None, y=None)对于条件数组中等价于True的位置，从第一个值或数组中取值进行替换，否则从第二个值或数组中取值进行替换。

x = np.array([1,2,3])
y = np.array([[0,0,3],[4,5,0]])
print(np.where(x%2==1,x,-x)) #[ 1 -2  3]
print(np.where(y, [10,20,30], [40,50,60])) #[[40 50 30] [10 20 60]]