Python如何通过布尔序列筛选嵌套列表得到对应目标子集

实现方案 #

方案1：列表推导式（最推荐，无需额外导入依赖）

直接在列表推导的判断条件里过滤即可，不需要先生成布尔序列，一行即可完成：

contests = [[5, 1], [2, 1], [1, 1], [8, 1], [10, 0], [5, 0]]
result = [item for item in contests if item[1] == 1]

执行后result的结果就是你需要的[[5, 1], [2, 1], [1, 1], [8, 1]]。

方案2：基于itertools.compress的一行写法

如果你坚持用compress，也可以直接把布尔序列的生成逻辑直接传入参数，不需要单独赋值变量：

from itertools import compress
contests = [[5, 1], [2, 1], [1, 1], [8, 1], [10, 0], [5, 0]]
result = list(compress(contests, [sublist[1] == 1 for sublist in contests]))

两种方案都能实现一行完成筛选，优先推荐方案1，语法更简洁，可读性也更高。

内容的提问来源于stack exchange，提问作者moth