moltiplicazione del vettore di matrice numpy

| | | | | | | | | |

Quando moltiplico due array numpy di dimensioni (nxn)*(nx 1), ottengo una matrice di dimensioni (nxn). Seguendo le normali regole di moltiplicazione delle matrici, è previsto un vettore (nx 1), ma semplicemente non riesco a trovare alcuna informazione su come farlo nel modulo Numpy di Python.

Il fatto è che non voglio per implementarlo manualmente per preservare la velocità del programma.

Il codice di esempio è mostrato di seguito:

a = np.array([[ 5, 1 ,3], [ 1, 1 ,1], [ 1, 2 ,1]]) b = np.array([1, 2, 3]) print a*b >> [[5 2 9] [1 2 3] [1 4 3]] 

Quello che voglio è:

stampa a*b >> ; [16 6 8]