본문 바로가기
파이썬/pyplot, seaborn

파이썬 기초 pyplot,seaborn - bar() , countplot() 막대그래프

by 큰고양2 2023. 8. 29.

pyplot bar 그래프

import matplotlib.pyplot as plt

로 pyplot을 임포트하고

plt.bar(x=인덱스, height = 값)

형태로 막대 그래프를 만들 수 있다

 

막대 그래프를 그릴때는 범주형 값을 사용하는데

x에는 범주 이름, height에는 값을 넣어주자

 

value_counts() 이용해서 그리기

temp = 데이터프레임['조회할 열'].value_counts()
plt.bar(x=temp.index.astype(str) , height = temp.values)

 

을 사용하면 해당 열의 범주 값을 막대 그래프로 그릴 수 있다

 

ex)

temp = data['OverTime'].value_counts()
plt.bar(x=temp.index.astype(str) , height = temp.values)

추가 옵션

plt.bar() 내부에 width 옵션으로 폭을 지정 할 수 있고

color = 으로 색, edgecolor = 으로 테두리 색 등을 지정 할 수 있다

 

 

seaborn - countplot()

 

sns.countplot(x=조회할 열)

으로 범주형 데이터의 막대그래프를 그릴 수 있다

plt.barplot과는 다르게 따로 value_counts()를 사용하지 않아도 알아서

범주별 숫자를 확인해서 그래프를 그려준다

+plt의 막대그래프를 plt.bar로 그려서  sns도 sns.barplot으로 착각 할 수 있는데

sns.barplot은 아예 다른 그래프 라는 점을 주의하자

 

ex)

sns.countplot(x=data['OverTime'])