Norm su riga in panda

Question

Ho un Dataframe panda con N colonne che rappresentano le coordinate di un vettore (ad esempio X, Y, Z, ma potrebbe essere più di 3D).

Vorrei aggregare il dataframe lungo le righe con una funzione arbitraria che combina le colonne, ad esempio la norma: (X^2 + Y^2 + Y^2).

voglio fare qualcosa di simile a ciò che è fatto here e here e here ma voglio tenerlo abbastanza generale che il numero di colonne può cambiare e si comporta come

DataFrame.mean(axis = 1) 

o

DataFrame.sum(axis = 1) 

Answer 1

ho trovato una soluzione più veloce di quello @elyase suggerito:

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

Answer 2

filtrare le colonne in base al nome

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

Stai cercando apply. Il tuo esempio sarà simile a questo:

>> 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 

Questo funziona per qualsiasi numero di dimensioni.

Answer 4

Numpy fornisce norma ... Usa:

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