python - 添加附加 numpy 数组

Question

我目前正在尝试将多个 Numpy 数组附加在一起。基本上，我想要做的是从 (1 x m) 矩阵(技术上是向量)开始，并以 (n x m) 矩阵结束。因此，从 n (1 x m) 矩阵(向量)到 1 (n x m) 矩阵(如果这有意义的话)。最终目标是使用 numpy.savetxt() 函数将矩阵写入 csv 文件，这样我最终会得到一个包含 n 列、长度为 m 的 csv 文件。

问题在于 numpy.append() 将向量一起附加到 (1 x 2m) 向量中。假设 a1 和 a2 是 Numpy 数组，每个数组都有 10000 个元素。我将使用append函数将a2附加到a1中，同时创建一个名为a的新数组，其中包含a1和a2。

a=np.append(a1, a2, axis=0)
a.shape
>>(20000,)

我想要的是形状是这样的

>>(2, 10000)

或者更一般地说

>>(n, m)

我该怎么办？请注意，我想继续将向量添加到数组中。感谢您的宝贵时间!

Answer 1

您可以使用numpy.column_stack的转置

例如:

import numpy as np

a=np.array([1,2,3,4,5])
b=np.array([9,8,7,6,5])
c=np.column_stack((a,b)).T

print c
>>> array([[1, 2, 3, 4, 5],
           [9, 8, 7, 6, 5]])

print a.shape,b.shape,c.shape
>>> (5,) (5,) (2, 5)

编辑:

您可以继续添加列，如下所示:

d=np.array([2,2,2,2,2])
c=np.column_stack((c.T,d)).T
print c
>>> array([[1, 2, 3, 4, 5],
           [9, 8, 7, 6, 5],
           [2, 2, 2, 2, 2]])
print c.shape
>>> (3, 5)