温馨提示×

NumPy怎么根据条件保留元素

小亿
91
2024-05-13 12:46:12
栏目: 编程语言
开发者测试专用服务器限时活动,0元免费领,库存有限,领完即止! 点击查看>>

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

亿速云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>

推荐阅读:NumPy怎么根据条件筛选元素

0