Python |添加具有不同形状的numpy数组

时间:2017-10-20 10:35:54

标签: python arrays numpy

我分别有两个形状(4,1)和(4)的numpy数组。我想以常规方式添加这两个数组,以获得具有四个元素的数组。例如,

A = np.array([[3],[5],[4],[2]])
B = np.array([3,5,4,2])

我希望C = A + B为np.array([6,10,8,4])(最好是形状(4,)。)但是,当我添加这两个数组(A和B)时,我得到一个4 * 4矩阵。

print A+B
[[  6.   8.   7.   5.]
 [  8.  10.   9.   7.]
 [  7.   9.   8.   6.]
 [  5.   7.   6.   4.]]

我缺少什么,如何实现我的目标?

0 个答案:

没有答案