如何将一行广播到Python Numpy中的列

Question

i有一个行向量R和一个列向量C。我想添加它们以创建一个数组a，其高度等于r和宽度等于C的大小等于C的大小如下：a [i，j] = r [i] c [j]

这样做的最有效方法是什么？

Answer 1

R + C[:, numpy.newaxis]

对我有用。

例如

import numpy as np
r = np.ones(5)
c = np.ones(4) * 2
r + c[:, np.newaxis]

给出

array([[ 3.,  3.,  3.,  3.,  3.],
       [ 3.,  3.,  3.,  3.,  3.],
       [ 3.,  3.,  3.,  3.,  3.],
       [ 3.,  3.,  3.,  3.,  3.]])