Python中按条件拆分列表元素：避免重复条件的优雅方案

当需要按特定条件拆分列表元素时，重复使用filter或列表推导式会导致两次遍历列表，复杂条件下还容易因重复编写逻辑出现错误。以下是更高效、易维护的实现方式：

一次遍历拆分法（推荐） #

编写通用的拆分函数，只需遍历列表一次，将元素按条件分配到两个结果列表中，条件逻辑仅需定义一次：

def split_by_condition(lst, condition):
    matches = []
    non_matches = []
    for item in lst:
        if condition(item):
            matches.append(item)
        else:
            non_matches.append(item)
    return matches, non_matches

示例：拆分奇数和偶数 #

def is_even(x):
    return x % 2 == 0

l = [0, 1, 2, 3, 4, 5, 6]
even, odd = split_by_condition(l, is_even)

print(even)  # 输出: [0, 2, 4, 6]
print(odd)   # 输出: [1, 3, 5]

这种方式的核心优势：

• 仅遍历列表一次，处理大型列表时性能更优
• 条件逻辑仅维护一处，避免重复编写导致的错误
• 函数通用性强，可适配任意拆分条件

简化版（列表推导式） #

如果列表规模较小、条件逻辑简单，也可以用列表推导式，但需注意这种方式会遍历列表两次：

even = [x for x in l if is_even(x)]
odd = [x for x in l if not is_even(x)]

不过对于复杂条件，仍推荐使用一次遍历的拆分函数，避免重复计算和逻辑不一致的问题。

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