Python:如何使数组的每个项重复，并将相同的项组合在一起

Question

我的数组是[A,B,C]，我想将它相乘，这样我就可以得到数组[A,A,A,B,B,B,C, C,C]

Answer 1

您可以链接 itertools.repeat 的结果:

import itertools
list(itertools.chain.from_iterable(itertools.repeat(x,3) for x in ["A","B","C"]))

结果:

['A', 'A', 'A', 'B', 'B', 'B', 'C', 'C', 'C']

这个解决方案最小化了循环和临时列表(没有创建)