3

例如,如果我有一个数组

[1, 2, 1]

我想得到一个长度是 4 倍的新数组

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

我该怎么做呢?

4

2 回答 2

8

这是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])
于 2013-09-06T16:21:58.303 回答
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]
于 2013-09-06T16:19:15.950 回答