例如,如果我有一个数组
[1, 2, 1]
我想得到一个长度是 4 倍的新数组
[1,1,1,1,2,2,2,2,1,1,1,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])
这是你想要做的吗?:
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]