Pandas: як додати стовпці на основі умови
Ви можете використовувати такий синтаксис, щоб підсумувати значення стовпця в pandas DataFrame на основі умови:
df. loc [df[' col1 '] == some_value , ' col2 ']. sum ()
У цьому підручнику наведено декілька прикладів практичного використання цього синтаксису з використанням наступних pandas DataFrame:
import pandas as pd #createDataFrame df = pd. DataFrame ({' team ': ['A', 'A', 'A', 'B', 'B', 'C'], ' conference ': ['East', 'East', 'East', 'West', 'West', 'East'], ' points ': [11, 8, 10, 6, 6, 5], ' rebounds ': [7, 7, 6, 9, 12, 8]}) #view DataFrame df team conference points rebounds 0 A East 11 7 1 A East 8 7 2 A East 10 6 3 B West 6 9 4 B West 6 12 5 C East 5 8
Приклад 1: додайте стовпець на основі умови
Наступний код показує, як знайти суму балів для рядків, де команда дорівнює “A”:
df. loc [df[' team '] == ' A ', ' points ']. sum () 29
Приклад 2: додайте стовпець на основі кількох умов
Наступний код показує, як знайти суму балів для ліній, де команда дорівнює «А» , а конференція дорівнює «Схід»:
df. loc [(df[' team '] == ' A ') & (df[' conference '] == ' East '), ' points ']. sum () 29
Приклад 3: додайте стовпець на основі однієї з кількох умов
Наступний код показує, як знайти суму балів для ліній, де команда дорівнює «A» або «B»:
df. loc [df[' team ']. isin ([' A ',' B ']), ' dots ']. sum () 41
Ви можете знайти більше посібників щодо панди на цій сторінці .