Pandas: 다른 열을 기반으로 열 값 추출

Pandas에서 query() 함수를 사용하여 다른 열의 값을 기반으로 한 열의 값을 추출할 수 있습니다.

이 함수는 다음 기본 구문을 사용합니다.

 df. query (" team=='A' ")[" points "]

이 특정 예에서는 팀 열이 A인 포인트 열에서 각 값을 추출합니다.

다음 예는 다음 pandas DataFrame에서 실제로 이 구문을 사용하는 방법을 보여줍니다.

 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 ': [11, 28, 10, 26, 6, 25, 29, 12]})

#view DataFrame
print (df)

  team position points
0 AG 11
1 AG 28
2 AF10
3AF 26
4 BG 6
5 BG 25
6 BF 29
7 BF 12

예시 1: 충족된 조건을 기준으로 컬럼 값 추출

다음 코드는 팀 열 값이 “A”인 포인트 열에서 각 값을 추출하는 방법을 보여줍니다.

 #extract each value in points column where team is equal to 'A'
df. query (" team=='A' ")[" points "]

0 11
1 28
2 10
3 26
Name: points, dtype: int64

이 함수는 팀 열의 해당 값이 “A”인 포인트 열의 4개 값을 반환합니다.

예시 2: 충족된 여러 조건 중 하나를 기반으로 열 값 추출

다음 코드는 팀 열 값이 “A” 이거나 위치 열 값이 “G”인 포인트 열에서 각 값을 추출하는 방법을 보여줍니다.

 #extract each value in points column where team is 'A' or position is 'G'
df. query (" team=='A' | position=='G' ")[" points "]

0 11
1 28
2 10
3 26
4 6
5 25
Name: points, dtype: int64

이 함수는 팀 열의 해당 값이 “A” 이거나 위치 열의 값이 “G”인 포인트 열의 6개 값을 반환합니다.

예시 3: 여러 조건이 충족된 컬럼 값 추출

다음 코드는 팀 열 값이 “A” 이고 위치 열 값이 “G”인 포인트 열에서 각 값을 추출하는 방법을 보여줍니다.

 #extract each value in points column where team is 'A' and position is 'G'
df. query (" team=='A' & position=='G' ")[" points "]

0 11
1 28
Name: points, dtype: int64

이 함수는 팀 열의 해당 값이 “A” 이고 위치 열의 값이 “G”인 포인트 열의 두 값을 반환합니다.

추가 리소스

다음 튜토리얼에서는 Pandas에서 다른 일반적인 작업을 수행하는 방법을 설명합니다.

Pandas: 조건에 따라 열을 선택하는 방법
Pandas: 여러 조건에 따라 행 삭제
Pandas: 다른 DataFrame을 기반으로 열 값 업데이트