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: 여러 조건을 기반으로 열 추가

다음 코드는 팀이 “A” 이고 컨퍼런스가 “동부”인 라인의 포인트 합계를 찾는 방법을 보여줍니다.

 df. loc [(df[' team '] == ' A ') & (df[' conference '] == ' East '), ' points ']. sum ()

29

예 3: 여러 조건 중 하나를 기반으로 열 추가

다음 코드는 팀이 “A” 또는 “B”와 같은 라인의 점수 합계를 찾는 방법을 보여줍니다.

 df. loc [df[' team ']. isin ([' A ',' B ']), ' dots ']. sum ()

41

이 페이지 에서 더 많은 팬더 튜토리얼을 찾을 수 있습니다.