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