numpy怎么筛选符合条件的行

1915
2024/2/19 9:41:29
栏目: 编程语言
开发者测试专用服务器限时活动,0元免费领,库存有限,领完即止! 点击查看>>

可以使用numpy的布尔索引来筛选符合条件的行。

布尔索引是一种使用布尔数组来选择数组中的元素的方法。我们可以通过创建一个布尔数组来指定筛选条件,然后将其与原数组进行索引,以获取符合条件的行。

以下是一个例子:

import numpy as np

# 创建一个示例数组
arr = np.array([[1, 2, 3],
                [4, 5, 6],
                [7, 8, 9]])

# 创建一个布尔数组,指定筛选条件
condition = arr > 5

# 使用布尔索引来筛选符合条件的行
filtered_arr = arr[condition]

print(filtered_arr)

输出结果为:

[6 7 8 9]

在上面的例子中,我们创建了一个二维数组arr,并创建了一个布尔数组condition来指定筛选条件,即arr中的元素大于5。然后,我们使用布尔索引arr[condition]来获得满足条件的元素,并将其保存在filtered_arr中。最后,我们打印出filtered_arr,即符合条件的行。

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

推荐阅读: NumPy数组拼接的方法有哪些