Norma wzdłuż rzędu w pandach

Question

Mam pandas Dataframe z N kolumnami reprezentującymi współrzędne wektora (na przykład X, Y, Z, ale może być więcej niż 3D).

Chciałbym agregować ramkę danych wzdłuż wierszy z dowolną funkcją, która łączy w sobie kolumny, na przykład normę: (X^2 + Y^2 + Y^2).

chcę zrobić coś podobnego do tego, co odbywa here i here i here ale chcę zachować go na tyle ogólne, że liczba kolumn można zmienić, a zachowuje się jak

DataFrame.mean(axis = 1) 

lub

DataFrame.sum(axis = 1) 

Answer 1

znalazłem szybsze rozwiązanie niż to, co @elyase zasugerował:

np.sqrt(np.square(df).sum(axis=1)) 

Answer 2

filtrować kolumny wg nazwy

cols = ['X','Y','Z'] 
df[cols].mean(axis=1) 
df[cols].sum(axis=1) 
df[cols].apply(lambda values: sum([v**2 for v in values]), axis=1) 

Answer 3

Szukasz apply. Twój przykład może wyglądać tak:

>> df = pd.DataFrame([[1, 1, 0], [1, 0, 0]], columns=['X', 'Y', 'Z']) 
    X Y Z 
0 1 1 0 
1 1 0 0 

>>> df.apply(lambda x: np.sqrt(x.dot(x)), axis=1) 
0 1.414214 
1 1.000000 
dtype: float64 

Działa to dla dowolnej liczby wymiarów.

Answer 4

Numpy przewiduje norma ... użyć:

np.linalg.norm(df[['X','Y','Z']].values,axis=1)