在Python / NumPy中，如何通过复制元素来填充数组？

Question

例如，如果我有一个数组

[1, 2, 1]

我希望得到一个长度为4倍的新数组

[1,1,1,1,2,2,2,2,1,1,1,1]

我该怎么做？

Answer 1

这是numpy.repeat的用途。

>>> x = numpy.array([1, 2, 1])
>>> numpy.repeat(x, 4)
array([1, 1, 1, 1, 2, 2, 2, 2, 1, 1, 1, 1])

Answer 2

这是你想要做的吗？：

def extend(lst, i): 
    return [e for e in lst for r in range(i)]
>>> extend([1,2,1], 4)
[1, 1, 1, 1, 2, 2, 2, 2, 1, 1, 1, 1]
>>> extend([1,2,1], 3)
[1, 1, 1, 2, 2, 2, 1, 1, 1]