通过使用NumPy将2个数组按元素数组明智地求和来形成矩阵 [英] Forming a matrix by summing 2 arrays element-array wise using NumPy

问题描述

假设我有以下NumPy数组:

Let's say I have the following NumPy arrays:

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

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

I would like to have a very efficient method (built-in if possible) to sum those vectors and have an output that looks like this:

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

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

So basically each column is the array j to which the k th element of i has been added (k = 0, 1, 2 in this case)

推荐答案

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

这篇关于通过使用NumPy将2个数组按元素数组明智地求和来形成矩阵的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！