python二维数组筛选的方法是什么

870
2024/6/11 10:36:09
栏目: 编程语言
开发者测试专用服务器限时活动,0元免费领,库存有限,领完即止! 点击查看>>

在Python中,我们可以使用列表推导式来筛选二维数组。例如,如果我们有一个二维数组matrix,想要筛选出满足某个条件的元素,可以使用以下方法:

matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

# 筛选出所有大于5的元素
filtered_matrix = [item for row in matrix for item in row if item > 5]

print(filtered_matrix)  # Output: [6, 7, 8, 9]

在上面的例子中,我们使用列表推导式来遍历二维数组matrix中的每个元素,并且只保留满足条件item > 5的元素。

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

推荐阅读: python如何获取接口返回值