python - 从 Python 中的字符串列表列表中获取连接字符串

Question

我有一个列表列表和一个分隔符字符串，如下所示:

lists = [
    ['a', 'b'],
    [1, 2],
    ['i', 'ii'],
]
separator = '-'

因此，我想要一个字符串列表，其中包含来自子列表中字符串的分隔符字符串:

result = [
    'a-1-i', 
    'a-1-ii', 
    'a-2-i', 
    'a-2-ii',
    'b-1-i', 
    'b-1-ii', 
    'b-2-i', 
    'b-2-ii',
]

结果中的顺序无关紧要。

我该怎么做？

Answer 1

from itertools import product
result = [separator.join(map(str,x)) for x in product(*lists)]

itertools.product 返回一个迭代器，该迭代器生成所提供的可迭代对象的笛卡尔积。我们需要在生成的元组上 map str，因为有些值是整数。最后，我们可以连接字符串化的元组并将整个内容放入列表推导式(如果处理大型数据集，则为生成器表达式，您只需要它进行迭代)。