[파이썬] python 피봇(pivot)사용하기, 인덱스 만들기(짧음)

원본의 table(농구선수의 정보)

 

피봇테이블을 마치 groupby처럼 사용할 수 있다.

pivot = data.pivot_table(
        ['Salary','Age'], #급여,나이
        index=['Position'], #포지션별로 급여,나이를 보겠다.
        aggfunc = 'mean' #평균
    )

pivot

피봇테이블을 사용하면 인덱스가 숫자로 나타나지않아서 활용할 때 힘드니까 인덱스를 넣어보자!

pivot2 = pivot.copy() #pivot의 복사본으로 pivot2만들자
pivot2.columns = pivot.columns.values

#index를 0~n까지, inplace=True 는 원본에 덮어씌우겠다는 뜻
#pivot의 index가 2개라면 reset_index 2번 반복해서 써주면 됌
pivot2.reset_index(level=0, inplace=True) 
pivot2

이렇게 짜잔하고 생김

 

이러면 하나하나 뽑을 때 포지션도 편하게 뽑을 수 있음.

 

print(pivot.iloc[0]) #기존의 pivot
print()
print(pivot2.iloc[0]) #인덱스를 넣은 pivot

밑에 결과로 position도 같이 나오는 모습