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