通過使用NumPy將2個數組按元素數組求和來形成矩陣

Question

假設我有以下NumPy數組：

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

我想有一個非常有效的方法（如果可能的話，可以內置）來對這些向量求和，並具有如下所示的輸出：

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

因此，基本上每一列都是添加了i的第k個元素的數組j（在這種情況下，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) 。