Python，itertools.combinations 结果到熊猫中的不同行

Question

假设我有一个列表：

list = [1, 2, 3, 4, 5]

我想得到成对的组合，它们是：

[1,2] [1,3] [1,4] [1,5] [2,3] [2,4] [2,5] [3,4] [3,5] [4,5]

然后我想将输出保存到 pandas 中的行。结果输出应如下所示：

     Combinations
0  [[1,2],[1,3],[1,4],[1,5]]
1  [[2,3],[2,4],[2,5]]
2  [[3,4], [3,5]]
3  [[4,5]]

我必须为包含 1000 个元素的列表执行此操作。任何帮助将不胜感激

Answer 1

您可以按它们的第一个元素对它们进行分组：

from itertools import combinations, groupby
from pandas import Series
from operator import itemgetter

combined = combinations(inputlist, 2)
series = Series(list(g) 
                for k, g in groupby(combined, key=itemgetter(0)))

演示：

>>> from itertools import combinations, groupby
>>> from pandas import Series
>>> from operator import itemgetter
>>> inputlist = [1, 2, 3, 4, 5]
>>> combined = combinations(inputlist, 2)
>>> Series(list(g) for k, g in groupby(combined, key=itemgetter(0)))
0    [(1, 2), (1, 3), (1, 4), (1, 5)]
1            [(2, 3), (2, 4), (2, 5)]
2                    [(3, 4), (3, 5)]
3                            [(4, 5)]
dtype: object