Calcul de la norme des colonnes en tant que vecteurs dans une matrice

Question

Je cherche la meilleure façon de calculer la norme des colonnes en tant que vecteurs dans une matrice. Mon code est en ce moment comme ça, mais je suis sûr qu'il peut être mieux (avec numpy peut-être?):

import numpy as np 
def norm(a): 
    ret=np.zeros(a.shape[1]) 
    for i in range(a.shape[1]): 
     ret[i]=np.linalg.norm(a[:,i]) 
    return ret 

a=np.array([[1,3],[2,4]]) 
print norm(a) 

qui retourne:

[ 2.23606798 5.  ] 

Merci.

Answer 1

Vous pouvez calculer la norme en utilisant ufuncs:

np.sqrt(np.sum(a*a, axis=0)) 

Answer 2

solution directe à l'aide numpy:

x = np.norm(a, axis=0)