Pandy: wzór na „jeśli wartość w kolumnie to”

Możesz użyć następującej składni w pandach, aby przypisać wartości do kolumny na podstawie wartości innej kolumny:

 df[' new '] = df[' col ']. map ( lambda x: ' new1 ' if ' A ' in x else ' new2 ' if ' B ' in x else '')

Ta konkretna składnia utworzy nową kolumnę o nazwie „new”, która przyjmie następujące wartości:

• new1 , jeśli wartość w col jest równa A.
• new2 , jeśli wartość w kolumnie jest równa B.
• Pusty ciąg znaków, jeśli wartość w kolumnie jest równa dowolnej innej wartości.

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

Przykład: użycie formuły „Jeśli wartość w kolumnie to” w Pandach

Załóżmy, że mamy następującą ramkę danych pand, która zawiera informacje o różnych koszykarzach:

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' team ': ['A', 'A', 'A', 'A', 'B', 'B', 'C', 'C'],
                   ' points ': [14, 22, 25, 34, 30, 12, 10, 18]})

#view DataFrame
print (df)

  team points
0 to 14
1 to 22
2 to 25
3 to 34
4 B 30
5 B 12
6 C 10
7 C 18

Załóżmy teraz, że chcemy utworzyć nową kolumnę o nazwie miasto , której wartości zależą od odpowiedniej wartości w kolumnie zespół .

W tym celu możemy użyć następującej składni:

 #create new column called city whose values depend on values in team column
df[' city '] = df[' team ']. map ( lambda x: ' Atlanta ' if ' A ' in x else ' Boston ' if ' B ' in x else '')

#view updated DataFrame                            
print (df)

  team points city
0 A 14 Atlanta
1 to 22 Atlanta
2 to 25 Atlanta
3 to 34 Atlanta
4 B 30 Boston
5 B 12 Boston
6 C 10         
7 C 18       

Ta konkretna składnia utworzyła nową kolumnę o nazwie miasto , która przyjmuje następujące wartości:

• Atlanta , jeśli wartość drużyny jest równa A.
• Boston , jeśli wartość zespołu jest równa B.
• Pusty ciąg znaków, jeśli wartość w zespole jest równa dowolnej innej wartości.

Zauważ, że w tym przykładzie użyliśmy pustego ciągu znaków po ostatniej instrukcji else , aby po prostu pozostawić wartości, które nie spełniają żadnych warunków, puste.

Dodatkowe zasoby

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

Pandy: pobierz indeks wierszy, których kolumna odpowiada wartości
Pandy: jak wybrać kolumny zawierające określony ciąg
Pandy: jak sprawdzić, czy kolumna zawiera ciąg znaków