Pandy: jak zastosować funkcję do każdego wiersza w dataframe

Możesz użyć następującej podstawowej składni, aby zastosować funkcję do każdego wiersza ramki DataFrame pandy:

 df[' new_col '] = df. apply ( lambda x: some function, axis= 1 )

Ta składnia stosuje funkcję do każdego wiersza ramki DataFrame pandy i zwraca wyniki w nowej kolumnie.

Poniższy przykład pokazuje, jak zastosować tę składnię w praktyce.

Przykład: Zastosuj funkcję do każdego wiersza w DataFrame

Załóżmy, że mamy następującą ramkę danych pandy:

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' A ': [5, 4, 7, 9, 12, 9, 9, 4],
                   ' B ': [10, 8, 10, 6, 6, 5, 9, 12]})

#view DataFrame
print (df)

    AB
0 5 10
1 4 8
2 7 10
3 9 6
4 12 6
5 9 5
6 9 9
7 4 12

Załóżmy teraz, że chcemy zastosować funkcję, która mnoży wartości w kolumnie A i kolumnie B, a następnie dzieli przez 2.

Możemy użyć następującej składni, aby zastosować tę funkcję do każdego wiersza ramki DataFrame:

 #create new column by applying function to each row in DataFrame
df[' z '] = df. apply ( lambda x: x[' A '] * x[' B '] / 2, axis= 1 )

#view updated DataFrame
print (df)

    AB z
0 5 10 25.0
1 4 8 16.0
2 7 10 35.0
3 9 6 27.0
4 12 6 36.0
5 9 5 22.5
6 9 9 40.5
7 4 12 24.0

Kolumna z wyświetla wyniki funkcji.

Na przykład:

• Pierwszy rząd: A * B / 2 = 5 * 10 / 2 = 25
• Drugi rząd: A * B / 2 = 4 * 8 / 2 = 16
• Trzeci rząd: A * B / 2 = 7 * 10 / 2 = 35

I tak dalej.

Możesz użyć podobnej składni z lambdą , aby zastosować dowolną funkcję do każdego wiersza ramki DataFrame pandy.

Dodatkowe zasoby

Poniższe samouczki wyjaśniają, jak wykonywać inne typowe operacje na pandach:

Jak zastosować funkcję do Pandas Groupby
Jak wykonać sumę GroupBy w Pandach
Jak korzystać z Groupby i Plot w Pandach