如何从多组元素中找到所有组合？

Question

我想从 3 组元素中形成所有可能的 6 种组合。组成如下：

• 从 A 组取 2 个 (a,b,c,d,e)
• 从 B 组中抽取 2 个 (f,g,h,i,j) 和
• 从 C 组中取 2 个 (k,l,m,n,o,p,q,r,s,t) 此外，在此之后，我想将其导出为 CSV 文件，如下所示：

Column 1    Column 2    Column 3    Column 4   Column 5    Column 6
   a            b        f            g         k           l 

等等……

ps。如果不允许使用字母，则可以编号。 我试过使用itertools 和列表，但仍然没有。我希望你能帮帮我。

Answer 1

这是你的排列代码：

# Print CSV header
print("Column 1,Column 2,Column 3,Column 4,Column 5,Column 6")

A = ['a', 'b', 'c', 'd', 'e']
B = ['f', 'g', 'h', 'i', 'j']
C = ['k', 'l', 'm', 'n', 'o', 'p', 'q', 'r', 's', 't']

for a1 in range(0, len(A)):
  for a2 in range(a1 + 1, len(A)):
    
    for b1 in range(0, len(B)):
      for b2 in range(b1 + 1, len(B)):
        
        for c1 in range(0, len(C)):
          for c2 in range(c1 + 1, len(C)):
            
            print(A[a1] + "," + A[a2] + "," + B[b1] + "," + B[b2] + "," + C[c1] + "," + C[c2])

它将为问题中指定的输入打印 4500 行。
它假定顺序无关紧要，因此如果出现(a, b)，则不会出现(b, a)。它也不会输出两次字母(a, a)不会出现。

如果您想使用双字母，请将 a1 + 1 替换为 a1（对于 b 和 c 也是如此）。
如果顺序很重要，（所以你想拥有(a, b) 和 (b, a)），将a1 + 1 替换为0（同样适用于b 和c）。