NumPy怎么根据条件筛选元素

在NumPy中，可以使用布尔索引来根据条件筛选数组元素。首先，我们需要创建一个条件数组，然后将这个条件数组作为索引传递给原始数组。例如，假设我们有一个一维数组arr，我们想要筛选出其中大于5的元素，可以按照以下步骤操作：

import numpy as np

arr = np.array([1, 6, 4, 9, 3, 8])

# 创建条件数组
condition = arr >
     5

# 使用条件数组筛选元素
filtered_arr = arr[condition]

print(filtered_arr)

这将输出 [6 9 8]，即原始数组中大于5的元素。您也可以将条件直接传递给索引：

filtered_arr = arr[arr >
     5]

这样也可以得到相同的结果。您可以根据需要更改条件，以满足不同的筛选需求。

声明：本文内容由网友自发贡献，本站不承担相应法律责任。对本内容有异议或投诉，请联系2913721942#qq.com核实处理，我们将尽快回复您，谢谢合作！