numpy里有一个非常神奇的函数叫做np.where()函数,下面我们来讲解一下这个函数的用法。
主要有两个用法。
首先传入y,然后如果满足条件condition,那么y中原来的数值变为x,否则不变化。
相当于[xv if c else yv for (c,xv,yv) in zip(condition,x,y)]
用法如下:
import numpy as np arr = np.array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]) arr1 = np.where(arr % 2 == 1, '奇数', arr) print(arr1)
输出:
['0' '奇数' '2' '奇数' '4' '奇数' '6' '奇数' '8' '奇数' '10']2. np.where(condition)
返回满足condition的索引。
用法如下:
import numpy as np arr = np.array([11, 3, 4, 5, 6, 7, 8, 9]) print(np.where(arr < 5))
输出:
(array([1, 2], dtype=int64),)



