如何复制嵌套列表中的元素？

Question

我有一个如下所示的嵌套列表：

[[1, 2, 3], [4, 5, 6], [7, 8, 9]]

我想将每个内部列表中的每个元素复制 3 并将它们并排放置：

[[1, 1, 1, 2, 2, 2, 3, 3, 3], [4, 4, 4, 5, 5, 5, 6, 6, 6], [7, 7, 7, 8, 8, 8, 9, 9, 9]]

这是我目前的解决方案：

from itertools import repeat

x2 = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

ls = []
for i in x2:
    ls.append([x for item in i for x in repeat(item, 3)])
ls
>>> [[1, 1, 1, 2, 2, 2, 3, 3, 3],
 [5, 5, 5, 6, 6, 6, 7, 7, 7],
 [9, 9, 9, 10, 10, 10, 11, 11, 11],
 [4, 4, 4, 8, 8, 8, 3, 3, 3]]

有什么方法可以让它更快？

Answer 1

这是一种解决方案：

ls=[sum([[i]*3 for i in k], []) for k in x2]

>>>print(ls)
[[1, 1, 1, 2, 2, 2, 3, 3, 3], [4, 4, 4, 5, 5, 5, 6, 6, 6], [7, 7, 7, 8, 8, 8, 9, 9, 9]]

Answer 2

正如您所要求的更快，以下是此处与其他地方介绍的不同可能方式之间的一些比较。

• 我使用repeat 使用您的方法，并添加了它的列表组合版本以匹配其他版本。
• 我用过@IoaTzimasversion using sum。
• 我使用@Steven Rumbalski的zip的版本来自Duplicate elements in a list
• 我使用了@Wololofrom the comments 的“纯”list-comp 解决方案。

结果（在我的机器上）是（按升序排列）：

using_zip         5.630
using_repeat_comp 5.818
using_sum         5.905
using_repeat      6.421
using_pure        6.878

（对我来说）令人惊讶的是，zip 解决方案实际上是最快的，而您的repeat 解决方案的 list-comp 版本第二。我会假设“纯”列表组合将是最快的，但实际上是最差的。我猜TIL...

---

还有一个结果（@Henry Tjhia提出的）：

using_sort        3.219

如您所见，明显更好，但我将其分开放置，因为它严重依赖于上述具体示例。因此，如果您想复制升序数字列表，那将是您的最佳选择。

---

生成它们的代码是：

import timeit
from itertools import repeat

x2 = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

def using_repeat():
    ls = []
    for i in x2:
        ls.append([x for item in i for x in repeat(item, 3)])
    return ls

def using_repeat_comp():
    return [[x for item in i for x in repeat(item, 3)] for i in x2]

def using_zip():
    return [[x for triple in zip(l, l, l) for x in triple] for l in x2]

def using_sum():
    return [sum([[i]*3 for i in k], []) for k in x2]

def using_pure():
    return [[val for val in sublist for _ in range(3)] for sublist in x2]

def using_sort():
    return [sorted(sub[:]*3) for sub in x2]

for func in [using_repeat, using_repeat_comp, using_zip, using_sum, using_pure, using_sort]:
    print(func.__name__, timeit.timeit(f"{func.__name__}()", f"from __main__ import {func.__name__}"))