如何求和两个向量以获得所有对的总和矩阵?

时间:2017-12-15 10:05:27

标签: python numpy

我想总结从两个向量中取出的所有可能的对,并将结果存储为numpy矩阵或2d数组。

两个载体:

v1 = np.array([1, 2, 3])
v2 = np.array([4, 5])

我需要这样的结果:

matrix([[5, 6],
        [6, 7],
        [7, 8]])

使用numpy是否有一种优雅的方法?

1 个答案:

答案 0 :(得分:0)

import numpy as np

v1 = np.array([1, 2, 3])
v2 = np.array([4, 5])

m = np.column_stack((v1+v2[0], v1+v2[1]))

输出:

[[5 6]
 [6 7]
 [7 8]]