如何以Pythonic且高效的方式创建排除指定元素的[1,..,n]序列

你原来用循环判断元素是否在排除列表里的方法虽然能用，但当排除的元素多了之后，in操作会变慢——因为列表的成员检查是挨个遍历的线性操作。下面给你几种更高效、更符合Python风格的写法：

1. 列表推导式+集合查找 #

把排除列表转成集合（集合的成员检查是O(1)的常数时间），再用列表推导式生成结果：

n = 4
exclude = [2, 3]
exclude_set = set(exclude)
result = [x for x in range(1, n+1) if x not in exclude_set]

这种写法既简洁又高效，是最常用的Pythonic实现方式。

2. 函数式风格：filter() + 集合 #

如果喜欢函数式写法，可以用filter()配合lambda表达式：

n = 4
exclude_set = set([2, 3])
result = list(filter(lambda x: x not in exclude_set, range(1, n+1)))

不过通常列表推导式的可读性和性能会比filter()略好一点，所以更推荐第一种写法。

3. 集合差集操作 #

直接把原范围和排除列表都转成集合，做差集后再转成有序列表：

n = 4
exclude_set = set([2, 3])
result = sorted(set(range(1, n+1)) - exclude_set)

这种写法代码极短，但要注意集合是无序的，所以需要用sorted()保证结果和原范围的顺序一致。如果不需要有序结果，可以去掉sorted()。

小提示 #

当排除的元素数量较多时，用集合的方法比直接用列表做in检查快很多；如果元素少，几种方法差异不大。日常开发里，列表推导式的写法是首选，兼顾简洁和性能。

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