如何将(3,)numpy向量转换为(2,2,3)矩阵?

时间:2018-04-20 20:11:05

标签: python numpy multidimensional-array numpy-broadcasting

我有一个形状为(3,)

的numpy数组
a = np.array([2,3,4])

如何将此向量转换为(2,2,3)数组的形状,即

np.array([[[2,3,4],[2,3,4]],[[2,3,4],[2,3,4]]])

1 个答案:

答案 0 :(得分:1)

np.resize会以这种方式滚动:

np.resize(a, (2,2,3))

或者只是乘以:

np.ones((2,2,3), dtype=a.dtype) * a