溫馨提示×

python二維數(shù)組篩選的方法是什么

小億
147
2024-05-23 10:36:09
欄目: 編程語言

在Python中,我們可以使用列表推導(dǎo)式來篩選二維數(shù)組。例如,如果我們有一個二維數(shù)組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]

在上面的例子中,我們使用列表推導(dǎo)式來遍歷二維數(shù)組matrix中的每個元素,并且只保留滿足條件item > 5的元素。

0