如何在numpy中生成矩阵向量

Question

我想在numpy中创建相同矩阵的向量(因此作为数组)。假设矩阵是:

w = np.array([[1,2],
[3,4],
[5,6]])

那么，我如何创建一个固定长度的向量，矩阵w在每个位置?

即:vector[0] = ... = vector[n] = w

Answer 1

不确定确切的预期输出，要创建额外的维度，您可以使用numpy.tile:

n = 3
vector = np.tile(w, (n, 1, 1))

NB。vector不是一个真正的(1D)矢量，它是一个3D数组

输出:

# vector
array([[[1, 2],
[3, 4],
[5, 6]],
[[1, 2],
[3, 4],
[5, 6]],
[[1, 2],
[3, 4],
[5, 6]]])
# vector[0]
array([[1, 2],
[3, 4],
[5, 6]])