在列表中重新放置str，其中n个字符串从0开始

Question

给定一个列表列表和一个整数，我需要用n个多字符串替换每个字符串。 其中n=给定的整数。 例如：

irv_to_stv_ballot([['NDP', 'CPC'], ['GREEN']], 3)
[['NDP0', 'NDP1', 'NDP2', 'CPC0', 'CPC1', 'CPC2'], ['GREEN0', 'GREEN1', 'GREEN2']]

Answer 1

也可以将itertools.repeat与enumerate()一起使用：

from itertools import repeat


def irv_to_stv_ballot(values, count):
    return [
        ["%s%d" % (e, i) for item in value for i, e in enumerate(repeat(item, count))]
        for value in values
    ]


print(irv_to_stv_ballot([["NDP", "CPC"], ["GREEN"]], 3))

输出：

[['NDP0', 'NDP1', 'NDP2', 'CPC0', 'CPC1', 'CPC2'], ['GREEN0', 'GREEN1', 'GREEN2']]