Norm entlang der Reihe in Pandas

Question

Ich habe ein Pandas Dataframe mit N Spalten, die die Koordinaten eines Vektors (zum Beispiel X, Y, Z, aber könnte mehr als 3D sein).

Ich möchte den Datenrahmen entlang der Zeilen mit einer beliebigen Funktion, die die Spalten kombiniert, zum Beispiel die Norm: (X^2 + Y^2 + Y^2).

Ich möchte etwas ähnliches tun, was here getan und here und here aber ich will es behalten allgemein genug, dass die Anzahl der Spalten ändern kann, und es verhält sich wie

DataFrame.mean(axis = 1) 

oder

DataFrame.sum(axis = 1) 

Answer 1

fand ich eine schnellere Lösung als das, was vorgeschlagen @elyase:

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

Answer 2

filtern, die Spalten nach Namen

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

Sie suchen nach apply. Ihr Beispiel würde so aussehen:

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

Dies funktioniert für eine beliebige Anzahl von Dimensionen.

Answer 4

Numpy bietet norm ... Verwendung:

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