通过使用NumPy对2个数组元素 - 数组进行求和来形成矩阵

问题描述 投票:2回答:2

假设我有以下NumPy数组:

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

我想有一个非常有效的方法(内置如果可能)来对这些向量求和并得到如下所示的输出:

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

所以基本上每列都是添加了第k个元素的数组j(在这种情况下k = 0,1,2)

python arrays numpy sum
2个回答
3
投票

使用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]])

3
投票

或者与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)

© www.soinside.com 2019 - 2024. All rights reserved.