溫馨提示×

numpy中where函數的用法是什么

小億
114
2024-01-25 10:02:09
欄目: 編程語言

numpy中的where函數用于根據給定的條件從原數組中選擇元素。它的用法如下:

numpy.where(condition[, x, y])

參數:

  • condition:需要滿足的條件,可以是一個布爾數組或布爾表達式。
  • x和y:可選參數,表示滿足條件的元素替換為x中的元素,不滿足條件的元素替換為y中的元素。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。

0
亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女