python - 字符串与特定变量字符的组合

Question

如何收集字符串的组合，其中某些字符(但不是全部)是可变的？

换句话说，我有一个输入字符串和一个字符映射表。字符映射表指定哪些字符是可变的，以及它们可以被替换为什么。然后该函数产生所有可能的组合。

为了将其放在上下文中，我正在尝试收集可能被 OCR 引擎误解的 OCR 输出字符串的可能变化。

输入示例:

"ABCD"

字符映射表示例:

dict(
    B=("X", "Z"),
    D=("E")
)

预期输出:

[
    "ABCD",
    "ABCE",
    "AXCD",
    "AXCE",
    "AZCD",
    "AZCE"
]

Answer 1

您可以使用 itertools.product :

>>> from itertools import product
>>> s = "ABCD"
>>> d = {"B": ["X", "Z"], "D": ["E"]}
>>> poss = [[c]+d.get(c,[]) for c in s]
>>> poss
[['A'], ['B', 'X', 'Z'], ['C'], ['D', 'E']]
>>> [''.join(p) for p in product(*poss)]
['ABCD', 'ABCE', 'AXCD', 'AXCE', 'AZCD', 'AZCE']

请注意，我创建了 d["D"]为了保持一致性，一个列表而不是简单的字符串。