[已解决]大佬们请帮我康康，一个关于排列组合的问题

TGgan

大佬们，问题是这样的：
我有两组数据，例如：
第一组数据 = [A1,B1,C1,D1]
第二组数据 = [A2,B2,C2,D2]
出于某种需要，我想弄第三组数据[A,B,C,D]，第三组数据中的A、B、C、D分别可能来自第一组或者第二组
现在我想弄一个列表出来，列表中，以列表的形式包含了第三组数据的所有可能性，请问有没有比较简单可行的方法 想学习学习 ，在此先谢谢各位大佬们啦！！！！！！

最佳答案

月排行榜 / 总排行榜

傻眼貓咪

2022-1-8 18:57:40

1. from itertools import combinations as c
   
2. A = ['a', 'b', 'c', 'd']
   
3. B = ['e', 'f', 'g', 'h']
   
4. arr = [list(i) for i in c(A+B, 4)]
   
5. print(arr)

复制代码

跳转到最佳答案楼层

傻眼貓咪

1. from itertools import combinations as c
   
2. A = ['a', 'b', 'c', 'd']
   
3. B = ['e', 'f', 'g', 'h']
   
4. arr = [list(i) for i in c(A+B, 4)]
   
5. print(arr)

复制代码

TGgan

傻眼貓咪 发表于 2022-1-8 18:57

谢谢！！学到了！！！！！！

TGgan

傻眼貓咪 发表于 2022-1-8 18:57

发现好像不太行，因为我的第三组数据里的A只能来自A1或者A2，这样排列出来效果不是那样的

傻眼貓咪

TGgan 发表于 2022-1-8 20:26
发现好像不太行，因为我的第三组数据里的A只能来自A1或者A2，这样排列出来效果不是那样的{:10_ ...

你的意思是 A1 或 A2 选其中一个，B1 或 B2 选其中一个，以此类推？

傻眼貓咪

TGgan 发表于 2022-1-8 20:26
发现好像不太行，因为我的第三组数据里的A只能来自A1或者A2，这样排列出来效果不是那样的{:10_ ...

1. A = ['A1', 'B1', 'C1', 'D1']
   
2. B = ['A2', 'B2', 'C2', 'D2']
   
3. C = [list(i) for i in zip(*[A, B])]
   
4. arr = [[a, b, c, d] for a in C[0] for b in C[1] for c in C[2] for d in C[3]]
   
5. print(arr)

复制代码

傻眼貓咪

TGgan 发表于 2022-1-8 20:26
发现好像不太行，因为我的第三组数据里的A只能来自A1或者A2，这样排列出来效果不是那样的{:10_ ...

假如数组 A 或 B 长度不一定（例如你的问题给定数据组 4 个元素为一组），则用：

1. # 任意长度数组 A 和 B 都可用此代码（但必须确保 A 和 B 长度一致）
   
2. A = ['A1', 'B1', 'C1', 'D1', 'E1']
   
3. B = ['A2', 'B2', 'C2', 'D2', 'E2']
   
4. C = [list(i) for i in zip(*[A, B])]
   
5. n = 2**(len(A))
   
6. m = n>>1
   
7. index = []
   
8. while m:
   
9.     key = 1
   
10.     temp = []
    
11.     while len(temp) != n:
    
12.         for i in range(m): temp.append(key)
    
13.         key = 0 if key else 1
    
14.     index.append(temp)
    
15.     m >>= 1
    
16. index = [list(elem) for elem in zip(*[each for each in index])]
    
17. arr = []
    
18. for i in index:
    
19.     temp = []
    
20.     for k, j in enumerate(i):
    
21.         temp.append(C[k][j])
    
22.     arr.append(temp)
    
23. 
    
24. print(arr)

复制代码

TGgan

傻眼貓咪 发表于 2022-1-8 21:43
假如数组 A 或 B 长度不一定（例如你的问题给定数据组 4 个元素为一组），则用：

谢谢谢谢大佬，我一定好好研究，哈哈哈哈，我在应用中确实是一组6个数据

myqf123

傻眼貓咪 发表于 2022-1-8 18:57

fmydedie

hornwong

PFBZSC

tianlai7266

tianlai7266

心驰神往

阿萨德按时

1molHF

1molHF

伽羅~

fmydedie

解决就好