python - 如何按列添加矩阵和向量？

Question

考虑以下因素:

>>> matrix = numpy.array([[1, 2, 3],
...                       [4, 5, 6],
...                       [7, 8, 9]])
>>> vector = numpy.array([10, 20, 30])
>>> matrix + vector
array([[11, 22, 33],
       [14, 25, 36],
       [17, 28, 39]])

这会将向量和矩阵逐行相加(即每一行都添加向量)。

如何按列执行相同的操作？结果应该是

>>> ???
array([[11, 12, 13],
       [24, 25, 26],
       [37, 38, 39]])

我知道我可以使用

>>> (matrix.T + vector).T
array([[11, 12, 13],
       [24, 25, 26],
       [37, 38, 39]])

但是我有很多这样的添加，并且使用这种双重换位将使代码变得非常难以阅读。有没有办法配置 ndarrays，使其沿第一个轴(而不是最后一个轴)执行加法？

Answer 1

使向量成为一列:

matrix + vector[:, None]