Numpy: Jak elementowo mnożyć dwa wektory, kształt (n, 1) i (n)?

Question

Mnożenie elementów w dwóch wektorach nie stanowi problemu, jeśli oba mają ten sam kształt, powiedzmy oba (n, 1) lub oba (n,). Jeśli jeden wektor ma kształt (n, 1) i drugi (n,), to jednak operator zwraca coś śmiesznego.

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

Powstała macierz nxn zawiera A_ {i, j} = a_i * b_j.

Jak mogę zatem wykonać mnożenie elementowe dla a i b?

Answer 1

Pokrój wektory w sposób, który sprawia, że ​​ich kształt dopasowanie:

a[:, 0] * b 

lub

a * b[:, None] 

Answer 2

dodać drugą oś do b do a i b mają takie same wymiary:

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

Ewentualnie przetransferuj a tak że transmisja działa:

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

(Wy pewnie chcą transpozycji wynik.)