使用NumPy如何拉伸、缩放或加倍数组？

Question

我想知道是否有一个numpy函数可以沿着特定的轴“拉伸”数组，就像下面这样：

a = [[1,2,3,4],[1,2,3,4]]

变成

a = [[1,1,2,2,3,3,4,4],[1,1,2,2,3,3,4,4]]

谢谢！

- timkado

Answer 1

import numpy as np
a = np.array([[1,2,3,4],[1,2,3,4]])

第一种可能性：

a.repeat(2, axis=1)

第二种方法是：

np.kron(a, [1,1])

两者都返回：

array([[1, 1, 2, 2, 3, 3, 4, 4],
       [1, 1, 2, 2, 3, 3, 4, 4]])

- eumiro