Pandas: 열이 값과 일치하는 행의 인덱스 가져오기

다음 구문을 사용하여 열이 특정 값과 일치하는 Pandas DataFrame의 행 인덱스를 가져올 수 있습니다.

 df. index [df[' column_name ']== value ]. tolist ()

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

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' team ': ['A', 'A', 'A', 'B', 'B', 'C', 'C', 'D'],
                   ' points ': [5, 7, 7, 9, 12, 9, 9, 4],
                   ' rebounds ': [11, 8, 10, 6, 6, 5, 9, 12]})

#view DataFrame
df

team points rebounds
0 to 5 11
1 to 7 8
2 to 7 10
3 B 9 6
4 B 12 6
5 C 9 5
6 C 9 9
7 D 4 12

예시 1: 열이 값과 일치하는 행의 인덱스 가져오기

다음 코드는 특정 값과 동일한 열이 있는 행의 인덱스를 가져오는 방법을 보여줍니다.

 #get index of rows where 'points' column is equal to 7
df. index [df[' points ']== 7 ]. tolist ()

[1, 2]

이는 인덱스 값 1 과 2 가 있는 행의 포인트 열에 “7” 값이 있음을 알려줍니다.

열이 특정 값보다 작거나 큰 행의 인덱스를 찾기 위해 보다 작음 및 보다 큼 연산자를 사용할 수도 있습니다.

 #get index of rows where 'points' column is greater than 7
df. index [df[' points '] > 7 ]. tolist ()

[3, 4, 5, 6]

이는 인덱스 값이 3 , 4 , 5 , 6 인 행의 포인트 열에 “7”보다 큰 값이 있음을 알려줍니다.

예시 2: 열이 문자열과 일치하는 행의 인덱스 가져오기

다음 코드는 열이 특정 문자열과 동일한 행의 인덱스를 가져오는 방법을 보여줍니다.

 #get index of rows where 'team' column is equal to 'B'
df. index [df[' team ']==' B ']. tolist ()

[3, 4]

이는 인덱스 값 3 과 4가 있는 행의 팀 열에 “B” 값이 있음을 알려줍니다.

예시 3: 여러 조건으로 행 인덱스 가져오기

다음 코드는 여러 열의 값이 특정 조건과 일치하는 행의 인덱스를 가져오는 방법을 보여줍니다.

 #get index of rows where 'points' is equal to 7 or 12
df. index [(df[' points ']== 7 ) | (df[' points ']== 12 )]. tolist ()

[1, 2, 4]

#get index of rows where 'points' is equal to 9 and 'team' is equal to 'B'
df. index [(df[' points ']== 9 ) & (df[' team ']==' B ')]. tolist ()

[3]

추가 리소스

Pandas DataFrame에서 셀 값을 얻는 방법
Pandas DataFrame에서 인덱스 이름을 바꾸는 방법
Pandas에서 이름별로 열을 정렬하는 방법