python - 如何从该表构建所有可能元组的列表？

Question

假设我有一组列定义:

Col1: value11 value12 value13
Col2: value21 value22 
Col3: value31 value32 value33
...

给定一些列的子集——2 个或更多——我想找到这些列的所有可能值。假设我选择上面的第 1 列和第 2 列:

(value11 value21)
(value11 value22)
(value12 value21)
(value12 value22)
(value13 value21)
(value13 value22)

如果我选择 2:3:

(value21 value31)
(value21 value32)
(value21 value33)
(value22 value31)
(value22 value32)
(value22 value33)

如果我选择了所有三个:

(value11 value21 value31)
(value11 value21 value32)
...

我在 python 中实现它，我想要一个快速算法来完成这个。我的输入是一个元组列表:(columnName, columnValueList)

有什么建议吗？

Answer 1

最好的方法是使用 itertools.product()。例如:

import itertools

group1 = ['a', 'b']
group2 = ['c', 'd']

print list(itertools.product(group1, group2))

#==> [('a', 'c'), ('a', 'd'), ('b', 'c'), ('b', 'd')]

此函数接受多个参数(即多列)。

有关 iterools.product() 的更多帮助，请参阅 this .