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에서 이름별로 열을 정렬하는 방법