Норма вдоль ряда в пандах

Question

У меня есть панда Dataframe с N столбцами, представляющими координаты вектора (например, X, Y, Z, но может быть больше 3D).

Я хотел бы объединить данные по строкам с произвольной функцией, которая объединяет столбцы, например, норма: (X^2 + Y^2 + Y^2).

Я хочу сделать что-то подобное тому, что делается here и here и here, но я хочу, чтобы это достаточно общее, что число столбцов может измениться, и он ведет себя как

DataFrame.mean(axis = 1) 

или

DataFrame.sum(axis = 1) 

Answer 1

я нашел более быстрое решение, чем то, что @elyase предложил:

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

Answer 2

фильтр столбцов по имени

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

Вы ищете apply. Ваш пример будет выглядеть так:

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

Это работает для любого количества измерений.

Answer 4

Numpy обеспечивает норму ... Использование:

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