Pandy: jak zamienić wartości w kolumnie na podstawie warunku

Możesz użyć następującej podstawowej składni, aby zastąpić wartości w kolumnie pandy DataFrame na podstawie warunku:

 #replace values in 'column1' that are greater than 10 with 20
df. loc [df[' column1 '] > 10, ' column1 '] = 20

Poniższe przykłady pokazują, jak używać tej składni w praktyce.

Przykład 1: Zamień wartości w kolumnie na podstawie warunku

Załóżmy, że mamy następującą ramkę DataFrame pand:

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' team ': ['A', 'A', 'A', 'A', 'B', 'B', 'B', 'B'],
                   ' position ': ['G', 'G', 'F', 'F', 'G', 'G', 'F', 'F'],
                   ' points ': [5, 7, 7, 9, 12, 13, 9, 14],
                   ' assists ': [3, 8, 2, 6, 6, 5, 9, 5]})

#view DataFrame
df

	team position points assists
0 A G 5 3
1 A G 7 8
2 A F 7 2
3 A F 9 6
4 B G 12 6
5 B G 13 5
6 B F 9 9
7 B F 14 5       

Możemy użyć poniższego kodu, aby zastąpić każdą wartość w kolumnie „punkty” większą niż 10 wartością 20:

 #replace any values in 'points' column greater than 10 with 20
df. loc [df[' points '] > 10, ' points '] = 20

#view updated DataFrame
df

	team position points assists
0 A G 5 3
1 A G 7 8
2 A F 7 2
3 A F 9 6
4 B G 20 6
5 B G 20 5
6 B F 9 9
7 B F 20 5

Należy pamiętać, że każda z trzech wartości w kolumnie „punkty” większa niż 10 została zastąpiona wartością 20.

Przykład 2: Zamień wartości w kolumnie na podstawie wielu warunków

Załóżmy, że mamy następującą ramkę DataFrame pand:

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' team ': ['A', 'A', 'A', 'A', 'B', 'B', 'B', 'B'],
                   ' position ': ['G', 'G', 'F', 'F', 'G', 'G', 'F', 'F'],
                   ' points ': [5, 7, 7, 9, 12, 13, 9, 14],
                   ' assists ': [3, 8, 2, 6, 6, 5, 9, 5]})

#view DataFrame
df

	team position points assists
0 A G 5 3
1 A G 7 8
2 A F 7 2
3 A F 9 6
4 B G 12 6
5 B G 13 5
6 B F 9 9
7 B F 14 5       

Możemy użyć poniższego kodu, aby zastąpić każdą wartość w kolumnie „pozycja”, w której punkty są mniejsze niż 10 lub asysty mniejsze niż 5, ciągiem „Bad”:

 #replace string in 'position' column with 'bad' if points < 10 or assists < 5
df. loc [(df[' points '] < 10) | (df[' assists '] < 5), ' position '] = ' Bad '

#view updated DataFrame
df

        team position points assists
0 A Bad 5 3
1 A Bad 7 8
2 A Bad 7 2
3 A Bad 9 6
4 B G 20 6
5 B G 20 5
6 B Bad 9 9
7 B F 20 5

Podobnie możemy użyć poniższego kodu, aby zastąpić każdą wartość w kolumnie „pozycja”, w której liczba punktów jest mniejsza niż 10 , a liczba podań mniejsza niż 5, ciągiem „Bad”:

 #replace string in 'position' column with 'bad' if points < 10 and assists < 5
df. loc [(df[' points '] < 10) & (df[' assists '] < 5), ' position '] = ' Bad '

#view updated DataFrame
df

        team position points assists
0 A Bad 5 3
1 A G 7 8
2 A Bad 7 2
3 A F 9 6
4 B G 12 6
5 B G 13 5
6 B F 9 9
7 B F 14 5

Zwróć uwagę, że w dwóch liniach, w których liczba punktów była mniejsza niż 10 , a asyst mniejsza niż 5, wartość „pozycji” została zastąpiona ciągiem „Zła”.

Dodatkowe zasoby

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

Jak wybierać wiersze na podstawie wielu warunków w Pandach
Jak utworzyć nową kolumnę na podstawie warunku w Pandach
Jak filtrować ramkę danych Pandas pod wieloma warunkami