Python中获取列表唯一组合的最快方法及数学依据探究

嘿，我来帮你搞定这个问题！要在Python里最快获取列表的唯一组合，**直接用标准库的itertools.combinations**绝对是首选——毕竟它是用C实现的，速度拉满，完全不用自己造轮子。

1. 基础用法：无重复元素的列表 #

如果你的列表里没有重复元素，直接调用itertools.combinations就行，它会生成所有长度为r的不重复组合（组合是无序的，比如(1,2)和(2,1)算同一个，只会生成一次）。

举个你提到的n=4，r=2的例子：

import itertools

my_list = [1, 2, 3, 4]
# 生成所有2元素组合，返回迭代器（省内存）
combinations_iter = itertools.combinations(my_list, 2)
# 转成列表查看结果
unique_combinations = list(combinations_iter)
print(unique_combinations)
# 输出：[(1, 2), (1, 3), (1, 4), (2, 3), (2, 4), (3, 4)]

这个结果的数量正好符合组合数公式：4!/(2!*(4-2)!)=6，完全没问题。

2. 处理含重复元素的列表 #

如果你的列表里有重复元素（比如[1,2,2,3]），直接用combinations会生成重复的组合（比如(1,2)会出现两次）。这时候我们需要先给列表去重，再生成组合：

my_list = [1, 2, 2, 3]
# 去重同时保持原有顺序（Python 3.7+可用）
unique_items = list(dict.fromkeys(my_list))
# 生成唯一组合
unique_combinations = list(itertools.combinations(unique_items, 2))
print(unique_combinations)
# 输出：[(1, 2), (1, 3), (2, 3)]

如果不需要保持顺序，也可以用list(set(my_list))去重，但集合会打乱元素顺序，按需选择就行。

3. 为什么不自己实现？ #

你可能好奇要不要自己写递归或循环来实现组合生成——不是不行，但效率差远了。比如下面是一个简单的递归实现：

def custom_combinations(lst, r):
    if r == 0:
        return [[]]
    if len(lst) < r:
        return []
    # 包含第一个元素的组合 + 不包含第一个元素的组合
    with_first = [[lst[0]] + combo for combo in custom_combinations(lst[1:], r-1)]
    without_first = custom_combinations(lst[1:], r)
    return with_first + without_first

这个方法在小数据量下没问题，但当n和r变大时（比如n=100，r=50），Python的递归/循环开销会让它比itertools.combinations慢几十甚至上百倍，完全没必要舍近求远。

4. IPython中的使用技巧 #

在IPython里调用的话，直接按常规方式写代码就行，比如：

In [1]: import itertools
In [2]: my_list = [1,2,3,4]
In [3]: list(itertools.combinations(my_list, 2))
Out[3]: [(1, 2), (1, 3), (1, 4), (2, 3), (2, 4), (3, 4)]

如果数据量很大，别直接转成list——迭代器可以逐个遍历处理，省内存：

In [4]: for combo in itertools.combinations(my_list, 2):
   ...:     print(combo)
   ...:
(1, 2)
(1, 3)
(1, 4)
(2, 3)
(2, 4)
(3, 4)

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