Pandas: 열 이름에서 열 인덱스를 얻는 방법
다음 방법을 사용하여 Pandas의 열 이름에서 열 인덱스 값을 가져올 수 있습니다.
방법 1: 열 이름에 대한 열 인덱스 가져오기
df. columns . get_loc (' this_column ')
방법 2: 여러 열 이름에 대한 열 인덱스 가져오기
cols = [' this_column ', ' that_column '] [df. columns . get_loc (c) for c in cols if c in df]
다음 예에서는 다음 pandas DataFrame에서 실제로 각 메서드를 사용하는 방법을 보여줍니다.
import pandas as pd #createDataFrame df = pd. DataFrame ({' store ': ['A', 'A', 'A', 'A', 'B', 'B', 'B', 'B'], ' sales ': [18, 10, 14, 13, 19, 24, 25, 29], ' returns ': [1, 2, 2, 3, 2, 3, 5, 4], ' recalls ': [0, 0, 2, 1, 1, 2, 0, 1]}) #view DataFrame print (df) store sales returns recalls 0 to 18 1 0 1 to 10 2 0 2 A 14 2 2 3 A 13 3 1 4 B 19 2 1 5 B 24 3 2 6 B 25 5 0 7 B 29 4 1
예시 1: 열 이름에 대한 열 인덱스 가져오기
다음 코드는 “returns”라는 열에 대한 열 인덱스 값을 가져오는 방법을 보여줍니다.
#get column index for column with the name 'returns' df. columns . get_loc (' returns ') 2
이름이 “return”인 열의 열 인덱스 값은 2 입니다.
참고 : Python에서는 열 인덱스 값이 0부터 시작합니다. 따라서 “return”은 DataFrame의 세 번째 열이므로 해당 인덱스 값은 2입니다.
예 2: 여러 열 이름에 대한 열 인덱스 가져오기
다음 코드는 DataFrame의 여러 열에 대한 열 인덱스 값을 가져오는 방법을 보여줍니다.
#define list of columns to get index for cols = [' store ', ' returns ', ' recalls '] #get column index for each column in list [df. columns . get_loc (c) for c in cols if c in df] [0, 2, 3]
결과에서 우리는 다음을 볼 수 있습니다:
- 이름이 “store”인 열의 열 인덱스 값은 0 입니다.
- 이름이 “return”인 열의 열 인덱스 값은 2 입니다.
- “reminders”라는 열의 열 인덱스 값은 3 입니다.
참고 : pandas get_loc() 함수에 대한 전체 문서는 여기에서 찾을 수 있습니다.
추가 리소스
다음 튜토리얼에서는 Pandas에서 다른 일반적인 작업을 수행하는 방법을 설명합니다.
Pandas DataFrame에서 셀 값을 얻는 방법
Pandas DataFrame에서 인덱스 이름을 바꾸는 방법
Pandas에서 이름별로 열을 정렬하는 방법