[pandas] 데이터프레임의 특정 컬럼을 행 인덱스로 설정하는 방법, set_index()

오늘은 판다스 데이터프레임의 특정 컬럼을 행 인덱스로 설정하는 방법에 대해 소개해드리겠습니다. 

 

아래와 같이 생긴 년월, 방문수라는 두 개의 컬럼을 가진 데이터프레임이 있다고 가정하겠습니다. 0, 1, 2, 3과 같은 정수형 위치 인덱스가 사용되고 있는 데이터프레임입니다. 

 

 

여기서 만약 "년월" 컬럼을 행 인덱스로 이동시키고 싶으면, 다음과 같이 코딩해주시면 됩니다. 만약 위 데이터프레임을 df라고 한다면, set_index() 메소드를 활용하여 특정 컬럼을 행 인덱스로 지정할 수 있습니다.

 

df = df.set_index(['년월'])

 

 

년월 컬럼이 인덱스 위치로 간 것을 확인하실 수 있습니다.

 

만약 다시 정수형 위치 인덱스로 바꾸고 싶다면, reset_index() 메소드를 활용하시면 됩니다. 그러면 다시 되돌아갑니다. 

 

df = df.reset_index()

 

 

데이터프레임 객체의 set_index() 메소드와 reset_index() 메소드가 어떤 일을 해주는지 이해되셨나요? ㅎㅎ