Question

为简单起见，我有4个向量（numpy.ndarray的类型），每个向量的大小均为256，我想创建一个新矩阵，其中每列对应于每个向量。因此，连接后，我将得到一个256 * 4矩阵

我尝试使用hstack或np.concatenate，但没有得到想要的结果。

预先感谢

Answer 1

使用vstack并转置T：

n = 256
a = np.arange(n) + 0
b = np.arange(n) + 1
c = np.arange(n) + 2
d = np.arange(n) + 3

x = np.vstack([a,b,c,d]).T
x

输出：

array([[  0,   1,   2,   3],
       [  1,   2,   3,   4],
       [  2,   3,   4,   5],
       ...,
       [253, 254, 255, 256],
       [254, 255, 256, 257],
       [255, 256, 257, 258]])

或者，或者，还有一种column_stack方法：

x = np.column_stack([a,b,c,d])

如果您想将其作为实际的matrix而不是ndarray：

m = np.matrix([a,b,c,d]).T

更新：为此，您可以将数组收集到一个列表中，然后将它们堆叠在一起：

n = 256
l = []
for i in range(4):
    l.append(np.arange(n) + i)

x = np.vstack(l).T
x

或者，也可以在循环的每一步将它们一一堆叠，然后在循环之后转置：

n = 256
x = np.arange(n)
for i in range(1,4):
    x = np.vstack([x, np.arange(n) + i])

x = x.T
x

如何在一个矩阵中串联多个向量（每个向量是新矩阵的一列）

1 个答案: