按子列表元素匹配過濾列表

Question

我有以下雙重嵌套的列表：

records = [[['Jack', 'male', 1],['Jack', 'male', 2],['Jack', 'male', 3]],[['Sally', 'female', 1],['Sally', 'female', 2],['Sally', 'female', 3]]]

我想根據最內層列表的第二個元素（通過索引）等於1的位置來過濾此列表。結果應如下所示：

records
[[['Jack', 'male', 1]],[['Sally', 'female', 1]]]

我認為某種形式的帶有條件的嵌套列表理解可能是我需要的，但我無法弄清楚。

Answer 1

您可以使用嵌套列表理解：

records = [[['Jack', 'male', 1],['Jack', 'male', 2],['Jack', 'male', 3]],[['Sally', 'female', 1],['Sally', 'female', 2],['Sally', 'female', 3]]]
final_records = [[i for i in b if i[-1] == 1] for b in records]

輸出：

[[['Jack', 'male', 1]], [['Sally', 'female', 1]]]