numpy中的where函數用于根據給定的條件從原數組中選擇元素。它的用法如下:
numpy.where(condition[, x, y])
參數:
返回值:
示例:
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
new_arr = np.where(arr > 3, arr, 0)
print(new_arr)
# 輸出:[0 0 0 4 5]
在上面的示例中,arr > 3 是一個布爾數組,滿足條件的元素為True,不滿足條件的元素為False。根據條件,將滿足條件的元素替換為arr中的對應元素,不滿足條件的元素替換為0,得到了新的數組new_arr。