通过使用 NumPy 对 2 个数组元素数组进行求和来形成矩阵

Question

假设我有以下 NumPy 数组：

i = array([2, 4, 5])
j = array([0, 1, 2])

我想要一个非常有效的方法（如果可能的话内置）来对这些向量求和并得到如下所示的输出：

[[2 4 5]
 [3 5 6]
 [4 6 7]]

所以基本上每一列都是数组 j 添加了 i 的第 k 个元素（在本例中为 k = 0, 1, 2）

Answer 1

使用numpy.add.outer.

>>> 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]])

Answer 2

或 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).