我想知道是否有一个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]]
提前致谢!
答案 0 :(得分:4)
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]])