numpy中where函數(shù)的用法是什么

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

numpy中的where函數(shù)用于根據(jù)給定的條件從原數(shù)組中選擇元素。它的用法如下：

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

參數(shù)：

  • condition：需要滿足的條件，可以是一個(gè)布爾數(shù)組或布爾表達(dá)式。
  • x和y：可選參數(shù)，表示滿足條件的元素替換為x中的元素，不滿足條件的元素替換為y中的元素。x和y可以是一個(gè)數(shù)組或標(biāo)量。

返回值：

  • 返回一個(gè)新的數(shù)組，形狀與原數(shù)組相同。

示例：

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 是一個(gè)布爾數(shù)組，滿足條件的元素為True，不滿足條件的元素為False。根據(jù)條件，將滿足條件的元素替換為arr中的對(duì)應(yīng)元素，不滿足條件的元素替換為0，得到了新的數(shù)組new_arr。

0 踩