numpy - 添加兩個數組以獲得矩陣

Question

如何找到兩個數組的所有和的矩陣？

與輸入

x1 = np.array([0, 1]) 
x2 = np.array([1,2,3]) 

我想這個輸出是這樣的：

[[1, 2, 3], [2, 3, 4]] 

Answer 1

您可以使用與NumPy的newaxis屬性：

x1[:, np.newaxis] + x2 

這是None的首字母縮寫：

In [2]: np.newaxis is None 
Out[2]: True 

這樣：

x1[:, None] + x2 

也適用。

Answer 2

您可以使用list comprehension這樣的例子：

x1 = np.array([0, 1]) 
x2 = np.array([1,2,3]) 

final = [[j+k for j in x2] for k in x1] 

# Or, maybe: 
# final = np.array([[j+k for j in x2] for k in x1]) 
# >>> array([[1, 2, 3], [2, 3, 4]]) 

打印（終）

輸出：

[[1, 2, 3], [2, 3, 4]]