Python动态列条件过滤列表：实现可变数量条件的循环

问题背景 #

现有一段Python代码用于过滤二维列表，需要将代码中固定的row[索引]替换为动态的列索引集合：column_to_check_for_include（指定需要检查包含关键词的列）和column_to_check_for_exclude（指定需要检查排除关键词的列）。这两个列表的长度不固定，可能是任意数量的列索引，需要适配这种动态变化的条件数量。

原固定索引的代码如下：

column_to_check_for_include = [
    1,
    2,
]  # 内容动态变化，可能是任意长度的列索引集合
column_to_check_for_exclude = [
    0,
    2,
    3,
]  # 内容动态变化，可能是任意长度的列索引集合


content = [
    ["John", "is", "cool", "sometimes"],
    ["Paul", "is", "bad", "everytime"],
    ["Ringo", "may be", "great", "sometimes"],
    ["Georges", "must be", "small", "yesterday"],
]
include_criteria = ["is", "must be"]
exclude_criteria = ["bad", "sometimes"]

for row in content:
    if all(
        exclude not in row[0]
        and exclude not in row[2]
        and exclude not in row[3]
        for exclude in exclude_criteria
    ):
        if any(
            include in row[1] or include in row[2]
            for include in include_criteria
        ):
            print(row)
            # 输出 ['Georges', 'must be', 'small', 'yesterday']

需要实现的逻辑是：遍历动态的列索引列表，自动生成对应数量的检查条件，替代原代码中的固定索引判断。

解决方案 #

核心是利用Python的all()和any()函数嵌套，实现动态列的条件检查：

• 排除规则：对每个排除关键词，确保所有指定的排除列中都不包含该关键词
• 包含规则：对每个包含关键词，只要任意一个指定的包含列中包含该关键词即可

修改后的完整代码：

column_to_check_for_include = [1, 2]
column_to_check_for_exclude = [0, 2, 3]

content = [
    ["John", "is", "cool", "sometimes"],
    ["Paul", "is", "bad", "everytime"],
    ["Ringo", "may be", "great", "sometimes"],
    ["Georges", "must be", "small", "yesterday"],
]
include_criteria = ["is", "must be"]
exclude_criteria = ["bad", "sometimes"]

for row in content:
    # 检查所有排除关键词都不在任何指定的排除列中
    exclude_pass = all(
        all(exclude not in row[col] for col in column_to_check_for_exclude)
        for exclude in exclude_criteria
    )
    if exclude_pass:
        # 检查至少一个包含关键词出现在任意指定的包含列中
        include_pass = any(
            any(include in row[col] for col in column_to_check_for_include)
            for include in include_criteria
        )
        if include_pass:
            print(row)

代码解释 #

1. 排除条件判断：

   • 内层all(exclude not in row[col] for col in column_to_check_for_exclude)：确保单个排除关键词不在所有指定的排除列中
   • 外层all(...) for exclude in exclude_criteria：确保所有排除关键词都满足上述条件

2. 包含条件判断：

   • 内层any(include in row[col] for col in column_to_check_for_include)：检查单个包含关键词是否存在于任意指定的包含列中
   • 外层any(...) for include in include_criteria：确保至少有一个包含关键词满足上述条件

运行代码后，输出结果与原代码一致：['Georges', 'must be', 'small', 'yesterday']，同时完全适配column_to_check_for_include和column_to_check_for_exclude的动态变化。

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