NumPy怎么根据条件筛选元素

numpy

975

2024/5/19 12:29:15

栏目: 编程语言

在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]

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

最新知识库