可以使用NumPy的np.where()
函數(shù)來根據(jù)條件保留元素。該函數(shù)接受三個(gè)參數(shù):條件、條件為True時(shí)的值、條件為False時(shí)的值。如果只傳入條件參數(shù),則np.where()
函數(shù)會(huì)返回滿足條件的元素的索引。
例如,如果想要保留數(shù)組arr
中大于0的元素,可以使用以下代碼:
import numpy as np
arr = np.array([1, -2, 3, -4, 5])
result = np.where(arr > 0, arr, 0)
print(result)
輸出結(jié)果為:
[1 0 3 0 5]
在上面的例子中,arr > 0
是條件,arr
是條件為True時(shí)的值,0
是條件為False時(shí)的值。因此,只有大于0的元素被保留下來,小于等于0的元素被替換為0。