假设我有以下NumPy数组:
i = array([2, 4, 5])
j = array([0, 1, 2])
我想有一个非常有效的方法(内置如果可能)来对这些向量求和并得到如下所示的输出:
[[2 4 5]
[3 5 6]
[4 6 7]]
所以基本上每列都是添加了第k个元素的数组j(在这种情况下k = 0,1,2)
>>> import numpy as np
>>> i = np.array([2, 4, 5])
>>> j = np.array([0, 1, 2])
>>>
>>> np.add.outer(j, i)
array([[2, 4, 5],
[3, 5, 6],
[4, 6, 7]])
或者与broadcasting
:
In [272]: i[:,None]+j
Out[272]:
array([[2, 3, 4],
[4, 5, 6],
[5, 6, 7]])
i[:,None]
制作了一个(3,1)
阵列,用(3,)
(或(1,3)
)播放制作(3,3)
。