multiplication de vecteurs de matrice numpy

| | | | | | | | | |

Lorsque je multiplie deux tableaux numpy de tailles (nxn)*(nx 1), j'obtiens une matrice de taille (nxn). En suivant les règles de multiplication matricielle normales, un vecteur (nx 1) est attendu, mais je ne trouve tout simplement aucune information sur la façon dont cela est fait dans le module Numpy de Python.

Le fait est que je ne veux pas pour l'implémenter manuellement afin de préserver la vitesse du programme.

Un exemple de code est présenté ci-dessous :

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

Ce que je veux c'est :

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