时间戳

首页 > 解决方案 > 从嵌套的字母列表中生成所有单词组合

python - 从嵌套的字母列表中生成所有单词组合

问题描述

问题是按字典顺序生成单词。Allword所以从生成Letters：

#we have a list containing lists which contains alphabets
#for eg.:
Letters=[['A','B','C'],['D','E','F'],['P','Q','R','S']]
#This is input and output is supposed to be like this
Allword=['ADP', 'ADQ', 'ADR', 'ADS', 'AEP', 'AEQ', 'AER', 'AES', 'AFP', 'AFQ', 'AFR', 'AFS', 'BDP', 'BDQ', 'BDR', 'BDS', 'BEP', 'BEQ', 'BER', 'BES', 'BFP', 'BFQ', 'BFR', 'BFS', 'CDP', 'CDQ', 'CDR', 'CDS', 'CEP', 'CEQ', 'CER', 'CES', 'CFP', 'CFQ', 'CFR', 'CFS']

我使用了迭代算法，它仅在 List Letters 的长度为 3 时才有用

    for i in Letters[0]:
        for j in Letters[1]:
            for k in Letters[2]:
                    b=i+j+k
                    Allword.append(b)

我正在尝试找到一种可以完成相同工作但不受长度限制的算法。我该怎么办？

标签： pythonlistalgorithmtraversal

解决方案

您需要内部列表的笛卡尔积。你有itertools.product这个。然后只需将结果元组映射str.join为获取字符串列表：

from itertools import product

list(map(''.join, product(*Letters)))
# ['ADP', 'ADQ', 'ADR', 'ADS', 'AEP', 'AEQ'...

推荐阅读