python - 按子列表元素匹配过滤列表

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]]]