Numpy: come moltiplicare in modo elementare due vettori, forma (n, 1) e (n,)?

Question

La moltiplicazione elementare di due vettori non è un problema se entrambi hanno la stessa forma, ad esempio entrambi (n, 1) o entrambi (n,). Se un vettore ha forma (n, 1) e l'altro (n,), tuttavia, l'operatore * restituisce qualcosa di divertente.

a = np.ones((3,1)) 
b = np.ones((3,)) 
print a * b 

Il risultante nxn matrice contiene A_ {i, j} = a_i * b_j.

Come posso eseguire la moltiplicazione elementare per a e b?

Answer 1

Fetta i vettori in modo da rendere la loro corrispondenza forma:

a[:, 0] * b 

o

a * b[:, None] 

Answer 2

Aggiungere un secondo asse di b a quella a e b avere le stesse dimensioni:

>>> a * b[:,np.newaxis] 
array([[ 1.], 
     [ 1.], 
     [ 1.]]) 

In alternativa, trasporre a in modo che la trasmissione funziona:

>>> a.T * b 
array([[ 1., 1., 1.]]) 

(si sarebbe probabilmente desidera trasporre il risultato.)