pandas의 pivot_table과 seaborn의 heatmap그래프를 이용한 출력예제(식사인원에 따른 팁의 금액 분석)[exam13_heatmap] > [파이썬]기초실습예제

import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns

#예제에 사용된 데이터는 exam11 과 동일한 데이터를 사용

#engine부분을 적지 않으면 엑셀관련 오류가 발생함
df = pd.read_excel('tips.xlsx', engine='openpyxl')

#print(df)

# index 요일을 그룹으로 묶어서 tip의 평균값을 보여줌,
# colums(x축에 보여줄 데이터),values(y축에 보여줄 데이터)
ls_data = pd.pivot_table(df, index=['day'], columns=['size'], values=['tip'])

#print(ls_data)

#피벗테이블의 데이터를 열지도로 보여줌
#annot를 True로 히야 열지도안에 데이터를 보여줌
#인원수가 많으면 팁이 많고,보통 1인당 1달러의 팁을 줌
sns.heatmap(ls_data, annot=True)
plt.show()