使用 numpy 拉伸、缩放或加倍数组?

tim*_*ado 3 python arrays numpy

我想知道是否有一个 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]]

提前致谢!

eum*_*iro 5

import numpy as np
a = np.array([[1,2,3,4],[1,2,3,4]])
Run Code Online (Sandbox Code Playgroud)

第一种可能性:

a.repeat(2, axis=1)
Run Code Online (Sandbox Code Playgroud)

或第二个:

np.kron(a, [1,1])
Run Code Online (Sandbox Code Playgroud)

两者都返回:

array([[1, 1, 2, 2, 3, 3, 4, 4],
       [1, 1, 2, 2, 3, 3, 4, 4]])
Run Code Online (Sandbox Code Playgroud)