python之将组合结果分组到带有子列表的列表中

我有列表a:

a = [0,1,2,3,4,5] 

然后我创建此列表的所有三元素组合。由于与二进制 3 位字符串形式的等价物相匹配，结果在 sublistach 中分组。例如，结果[0,4,5]对应序列001，因为每个偶数对应0，奇数对应1。

我在注释中使用的代码:

import itertools as it 
import itertools 
 
# I create three-element combinations of zeros and ones 
combinations_3bit = list(map(list, itertools.product([0,1], repeat=3))) 
# output: [[0, 0, 0], [0, 0, 1], [0, 1, 0], [0, 1, 1], [1, 0, 0], [1, 0, 1], [1, 1, 0], [1, 1, 1]] 
 
# I create a list of strings with values from the list `3bit_combinations` 
a=[] 
for i in range(len(combinations_3bit)): 
    a2 =''.join(map(str, [1 if x%2 else 0 for x in combinations_3bit[i]])) 
    a.append(a2) 
# output: ['000', '001', '010', '011', '100', '101', '110', '111']  
 
# I remove repetitions in pairs of type 001 with 100, 011 with 110, etc. 
combinations_3bit_without_repetition = [v for k, v in enumerate(a) if v[::-1] not in a[:k]]  
# output: ['000', '001', '010', '011', '101', '111']  
 
b = [0,1,2,3,4,5] 
 
good = [] 
for i in range(len(combinations_3bit_without_repetition)): 
    c=[] 
    for u in it.combinations(b, 3):        
        u1 = list(u)                   
        y =''.join(map(str, [1 if x%2 else 0 for x in u1])) 
        if y == combinations_3bit_without_repetition[i]: 
           c.append(u1) 
    good.append(c)  
 
# output: [[[0, 2, 4]], [[0, 2, 3], [0, 2, 5], [0, 4, 5], [2, 4, 5]], [[0, 1, 2], [0, 1, 4], [0, 3, 4], [2, 3, 4]], [[0, 1, 3], [0, 1, 5], [0, 3, 5], [2, 3, 5]], [[1, 2, 3], [1, 2, 5], [1, 4, 5], [3, 4, 5]], [[1, 3, 5]]] 

能否更好、更经济地解决这个问题？ 因为上面的解决方案似乎是“周围”，例如函数 it.combinations 在列表 combinations_3bit_without_repetition 中的每个索引 i 之后返回所有可能组合，然后条件仅筛选匹配的组合。在大列表的情况下，这个解决方案很弱；)

请您参考如下方法:

有一种更好的方法来生成您需要的二进制字符串:

import itertools 
 
strings = ['{0:03b}'.format(i) for i in range(8)] 
b = [0,1,2,3,4,5] 
combinations = [list(x) for x in itertools.combinations(b, 3)] 
 
dct = {} 
 
for x in combinations: 
  y = ''.join(str(j%2) for j in x) 
  if y in dct: 
    dct[y].append(x) 
  else: 
    dct[y] = [x] 
 
print(dct) 

输出:

{'010': [[0, 1, 2], [0, 1, 4], [0, 3, 4], [2, 3, 4]], '011': [[0, 1, 3], [0, 1, 5], [0, 3, 5], [2, 3, 5]], '001': [[0, 2, 3], [0, 2, 5], [0, 4, 5], [2, 4, 5]], '000': [[0, 2, 4]], '101': [[1, 2, 3], [1, 2, 5], [1, 4, 5], [3, 4, 5]], '100': [[1, 2, 4]], '110': [[1, 3, 4]], '111': [[1, 3, 5]]} 

检查这是否满足您的需要。它创建一个字典，其中每个键是一个长度为 3 的二进制字符串，每个值是一个与二进制字符串匹配的组合数组。