技术问答NumPy怎么根据条件保留元素 - 扬州沐宇科技

扫描二维码下载沐宇APP

微信扫码使用沐宇小程序

沐宇首页互联网技术知识问答 NumPy怎么根据条件保留元素

扬州沐宇科技

2024-05-13 12:46:12

numpy

可以使用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。