numpy.where()的用法
1. np.where(condition, x, y)
满足条件(condition),输出x,不满足输出y。
arr = np.arange(10) out = np.where(arr%2 == 1,-1,arr) print(out) 4 >>> [ 0 -1 2 -1 4 -1 6 -1 8 -1]
2. np.where(condition)
只有条件 (condition),没有x和y,则输出满足条件 (即非0) 元素的坐标。这里的坐标以tuple的形式给出,通常原数组有多少维,输出的tuple中就包含几个数组,分别对应符合条件元素的各维坐标。
a = np.array([1,3,5,7,9]) print(np.where(a>5)) print(a[np.where(a>5)]) >>> (array([3, 4], dtype=int64),) [7 9]
3.给where()函数传递一个条件数组和两个值或数组,对于条件数组中等价于True的位置,从第一个值或数组中取值进行替换,否则从第二个值或数组中取值进行替换。
x = np.array([1,2,3,4,5,6]) y = np.array([[1,0,3],[4,5,0]]) print(np.where(x%2 == 1,x,-x)) print(np.where(y%2 == 1,1,-1)) print(np.where(y,[30,40,50],[60,70,80]))
#y数组的第一行,1等价于true,取第二组参数30,0等价于false,取第三组参数60,3等价于true,取第二组参数50 >>> [ 1 -2 3 -4 5 -6] [[ 1 -1 1] [-1 1 -1]] [[30 70 50] [30 40 80]]