Question

我试图从给定特定键的字典中的值中获取所有可能的字符串。例如，＆＃39; A＆＃39;可能意味着＆＃39; aaa＆＃39;，＆＃39; aba＆＃39;或者＆＃39; aac＆＃39;和＆＃39; B＆＃39;可能意味着＆＃39; bbb＆＃39;＆＃39; bab＆＃39;或＆＃39; bbc＆＃39;等等。

我已经给出了一个输出示例，其中显示了可能字符串的大多数组合

import itertools
in_ = 'ABC'
D = {'A':['aaa','aba','aac'],'B':['bbb','bab','bbc'],'C':['ccc','cac','ccb']}

#out_ = ['aaabbbccc','ababbbccc','aacbbbccc','aaababccc','aaabbcccc','aaabbbcac','aaabbbccb'...]

我开始编写代码，但它开始变得非常混乱，我觉得有一种方法可以使用itertools或pythonic来实现这个目标

output = []
for char in in_:
    out_string = ''
    while char:
        for v in D[char]:
            while v:
                for char2 in in_:
                    out_string
#not pythonic . . .

Answer 1

嗯，你已经在那里导入了itertools。让我们来使用吧！我们想要采用笛卡尔积D['A'] × D['B'] × D['C']，因此我们使用itertools.product来做到这一点。

import itertools

in_ = 'ABC'
D = {'A':['aaa','aba','aac'],'B':['bbb','bab','bbc'],'C':['ccc','cac','ccb']}

iterables = [D[character] for character in in_]
out_ = [''.join(tup) for tup in itertools.product(*iterables)]

现在，out_是：

['aaabbbccc', 'aaabbbcac', 'aaabbbccb', 'aaababccc', 'aaababcac', 'aaababccb', 
 'aaabbcccc', 'aaabbccac', 'aaabbcccb', 'ababbbccc', 'ababbbcac', 'ababbbccb', 
 'abababccc', 'abababcac', 'abababccb', 'ababbcccc', 'ababbccac', 'ababbcccb', 
 'aacbbbccc', 'aacbbbcac', 'aacbbbccb', 'aacbabccc', 'aacbabcac', 'aacbabccb', 
 'aacbbcccc', 'aacbbccac', 'aacbbcccb']

这是你想要的结果吗？

给定输入字符串的所有可能的字典值组合。蟒蛇

1 个答案: