過濾 Python 中元組的嵌套列表

Question

我在 python [[(1,2),(3,7)],[(5,22)],[(4,-1),(-3,34),(41,2)]]我想用“過濾器”過濾子列表所以條件是子列表中每個元組的所有第一個元素的總和是否大於5，如果不刪除子列表。

也許你知道怎么做？ （沒有包或進口）謝謝

Answer 1

你可以使用列表理解

a = [[(1,2),(3,7)],[(5,22)],[(4,-1),(-3,34),(41,2)]]
b = [x for x in a if sum(y for y,_ in x) > 5]

根據 azro 的評論進行編輯。

x 是 a 中的每個子數組，總和取每個子數組的第一個元素，如果總數大於 5，則 x 包含在 b 中。

Answer 2

您可以使用帶有過濾條件的嵌套列表推導，如下所示：

sub_lists = [[(1,2),(3,7)],[(5,22)],[(4,-1),(-3,34),(41,2)]]

filtered_sub_lists = [
    sub_list for sub_list in sub_lists if sum(values[0] for values in sub_list) > 5
]

print(filtered_sub_lists)
>>> [[(4, -1), (-3, 34), (41, 2)]]