Come usare numpy per aggiungere due elementi in una matrice e produrre una matrice?

Question

I codici pitone nativi sono come questo:

>>> a=[1,2,3,4,5,6] 
>>> [[i+j for i in a] for j in a] 
[[2, 3, 4, 5, 6, 7], 
[3, 4, 5, 6, 7, 8], 
[4, 5, 6, 7, 8, 9], 
[5, 6, 7, 8, 9, 10], 
[6, 7, 8, 9, 10, 11], 
[7, 8, 9, 10, 11, 12]] 

Tuttavia, devo usare NumPy per fare questo lavoro come l'array è molto grande. Qualcuno ha idee su come fare lo stesso lavoro in numpy?

Answer 1

Molti operatori binari NumPy hanno un metodo outer che può essere usato per formare l'equivalente di una moltiplicazione (o in questo caso, oltre) tabella:

In [260]: import numpy as np 
In [255]: a = np.arange(1,7) 

In [256]: a 
Out[256]: array([1, 2, 3, 4, 5, 6]) 

In [259]: np.add.outer(a,a) 
Out[259]: 
array([[ 2, 3, 4, 5, 6, 7], 
     [ 3, 4, 5, 6, 7, 8], 
     [ 4, 5, 6, 7, 8, 9], 
     [ 5, 6, 7, 8, 9, 10], 
     [ 6, 7, 8, 9, 10, 11], 
     [ 7, 8, 9, 10, 11, 12]])