[파이썬] 데이터프레임(dataframe) 인덱스 설정 / 해제 - set_index, reset_index

 

데이터프레임을 다룰 때, 인덱스를 설정하는 일이 참 많은데요,

인덱스를 설정하려면 set_index, 해제하려면 reset_index 라고 입력하면 됩니다.

아래의 코드를 보며 살펴보도록 하겠습니다.

 

 

1. set_index : 인덱스 설정하기

set_index() 함수를 이용하여 인덱스를 설정할 수 있는데요, 인덱스로 설정할 칼럼 이름을 괄호 안에 입력해주면 된답니다.  

import pandas as pd
info = pd.DataFrame({'item' : ['coffee', 'tea', 'juice' ], 
                     'price' : [3000, 4000, 5000], 
                     'rating' : [4.0, 3.5, 3.7]})
info

 

	item	price	rating
0	coffee	3000	4.0
1	tea	4000	3.5
2	juice	5000	3.7

 

만약 위와 같은 데이터프레임이 있을 때, 'item' 이라는 칼럼을 인덱스로 설정해주고 싶으면 아래 코드처럼 입력하면 됩니다.

 

info2 = info.set_index('item')
info2

 

	price	rating
item		
coffee	3000	4.0
tea	4000	3.5
juice	5000	3.7

 

출력된 결과를 보면 'item' 컬럼이 인덱스가 된 것을 확인할 수 있습니다.

 

 

2. reset_index : 인덱스 해제하기

한편, 설정한 인덱스를 해제하고 싶다면 reset_index() 를 사용하면 되는데요, 아래의 코드처럼 입력하면 됩니다.

 

info2 = info2.reset_index()
info2

 

	item	price	rating
0	coffee	3000	4.0
1	tea	4000	3.5
2	juice	5000	3.7

 

출력된 결과를 보면 인덱스가 다시 해제되었음을 알 수 있습니다.