用矩阵 B 附加矩阵 A

Posted 2023-03-11

【中文标题】用矩阵 B 附加矩阵 A

【英文标题】：Appending matrix A with matrix B

【发布时间】：2017-04-20 13:09:43

【问题描述】：

假设我有两个矩阵A 和B。例如，

A = numpy.zeros((5,5))

B = np.eye(5)

有没有办法追加A和B？

【参考方案1】：

听起来你在寻找np.hstack：

>>> import numpy as np
>>> a = np.zeros((5, 5))
>>> b = np.eye(5)
>>> np.hstack((a, b))
array([[ 0.,  0.,  0.,  0.,  0.,  1.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  1.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  1.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  1.,  0.],
       [ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  1.]])

如果您想将它们向下堆叠，np.vstack 将起作用：

>>> np.vstack((a, b))
array([[ 0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.],
       [ 1.,  0.,  0.,  0.,  0.],
       [ 0.,  1.,  0.,  0.,  0.],
       [ 0.,  0.,  1.,  0.,  0.],
       [ 0.,  0.,  0.,  1.,  0.],
       [ 0.,  0.,  0.,  0.,  1.]])

【讨论】：

或 np.stack((A,B), axis=n) 用于 0,1,2 创建 3d 数组。 :) 两个 5x5 数组可以通过至少 5 种方式连接。