linalg.norm nie przyjmuje argumentu osi

Question

Używam Pythona 3 w Pyzo. Czy mógłbyś mi powiedzieć, dlaczego funkcja linalg.norm nie rozpoznaje argumentu osi.

ten kod:

c = np.array([[ 1, 2, 3],[-1, 1, 4]]) 
d=linalg.norm(c, axis=1) 

zwraca błąd:

TypeError: norm() got an unexpected keyword argument 'axis'

Answer 1

linalg.norm nie akceptuje axis argument. Można obejść to z:

np.apply_along_axis(np.linalg.norm, 1, c) 
# array([ 3.74165739, 4.24264069]) 

Albo być szybciej wdrożyć się z:

np.sqrt(np.einsum('ij,ij->i',c,c)) 
# array([ 3.74165739, 4.24264069]) 

Do czasu:

timeit np.apply_along_axis(np.linalg.norm, 1, c) 
10000 loops, best of 3: 170 µs per loop 

timeit np.sqrt(np.einsum('ij,ij->i',c,c)) 
100000 loops, best of 3: 10.7 µs per loop 

Answer 2

Na numpy wersjach poniżej 1.8linalg.norm nie bierze axis argumentu można użyć np.apply_along_axis, aby uzyskać pożądany efekt, jak wskazano przez Warren Weckesser w komentarzu do pytania.

import numpy as np 
from numpy import linalg 

c = np.array([[ 1, 2, 3],[-1, 1, 4]]) 

d = np.apply_along_axis(linalg.norm, 1, c) 

Wynik:

>>> d 
array([ 3.74165739, 4.24264069])