np.where用法举例

np.where是NumPy库中的一个函数，用于根据给定的条件返回符合条件的元素的索引或值。它的用法如下：

1. 返回符合条件的元素的索引：

   import numpy as np
   
   arr = np.array([1, 2, 3, 4, 5])
   indices = np.where(arr > 3)
   print(indices)

输出：

   (array([3, 4]),)

这的条件是arr 3，np.where返回了满足条件的元素的索，即3和4。

2. 返回符合条件的元素值：

   import numpy as np
   
   arr = np.array([1, 2, 3, 4, 5])
   values = np.where(arr > 3, arr, 0)
   print(values)

输出：

   [0 0 0 4 5]

这里的条件是arr > 3，np.where返回了满足条件的元素的值，不满足条件的元素用0代替。

3. 多个条件的使用：

   import numpy as np
   
   arr = np.array([1, 2, 3, 4, 5])
   values = np.where((arr > 2) & (arr < 5), arr, -1)
   print(values)

输出：

   [-1 -1  3  4 -1]

这里的条件是arr > 2和arr < 5，np.where返回了同时满足两个条件的元素的值，不满足条件的元素用-1代替。