python中np.where的用法
np.where() 是 NumPy 庫中的一個函數,用于根據條件從多個選擇中選擇元素。
它的一般語法如下:
np.where(condition, x, y)
參數說明:
- condition:表示一個條件表達式,可以是布爾值數組或條件語句。
- x:當 condition 中的元素為 True 時,將選擇這些位置上的對應元素放入結果數組中。
- y:當 condition 中的元素為 False 時,將選擇這些位置上的對應元素放入結果數組中。
返回值:
- 根據 condition 條件的結果,返回一個新的數組。
以下是 np.where() 的一些常見用法示例:
1.根據條件選擇元素:
import numpy as np arr = np.array([1, 2, 3, 4, 5]) condition = arr > 3 result = np.where(condition, arr, 0) print(result) # 輸出: [0 0 0 4 5]
在上面的示例中,如果 arr 數組中的元素大于 3,那么結果數組對應位置的元素就是 arr 中的對應元素;否則,結果數組對應位置的元素就是 0。
2.使用條件語句:
import numpy as np arr1 = np.array([1, 2, 3]) arr2 = np.array([4, 5, 6]) condition = True result = np.where(condition, arr1, arr2) print(result) # 輸出: [1 2 3]
在上面的示例中,由于條件 condition 為 True,所以結果數組中的元素來自 arr1 數組;若 condition 為 False,則結果數組中的元素來自 arr2 數組。
np.where() 函數在數據處理和條件選擇方面非常有用,可以根據不同的條件進行靈活的元素選擇和替換操作。
到此這篇關于python中np.where的用法的文章就介紹到這了,更多相關python中np.where 內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家!
相關文章
PyTorch函數torch.cat與torch.stac的區(qū)別小結
Pytorch中常用的兩個拼接函數torch.cat() 和 torch.stack(),本文主要介紹了這兩個函數的用法加區(qū)別,具有一定的參考價值,感興趣的可以了解一下2023-09-09