如何在Python NumPy中向行广播一行?

时间:2017-12-20 22:15:58

标签: numpy numpy-broadcasting

我有一个行向量R和一个列向量C.我想添加它们来创建一个数组A,其高度等于R的大小,宽度等于C的大小,如下所示:A [i,j] = R [i] + C [j]

最有效的方法是什么?

1 个答案:

答案 0 :(得分:2)

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.]])
相关问题