BLOG/데이터분석
[데이터분석] matplotlib : 수평 막대그래프 간단하게 만들기
알디노
2021. 3. 10. 14:54
이전 포스팅에서는 matplotlib을 이용해서 막대그래프를 그리는 방법에 대해 살펴보았는데요,
이번에는 수평 막대그래프를 그리는 방법에 대해서 살펴보도록 하겠습니다.
수평 막대그래프를 그리는 방법은 수직 막대그래프와 코드 몇 줄 빼고 거의 동일한데요,
아래 코드와 출력된 그래프를 한 번 보도록 하겠습니다.
import matplotlib.pyplot as plt
import numpy as np
y = np.arange(3)
item = ['coffee', 'tea', 'ade']
values = [2000, 4000, 5000]
plt.barh(y, values)
plt.yticks(y, item)
plt.grid(True)
plt.show()
수직 막대그래프를 그리는 방법과 다른 점이 있다면 plt.barh() 와 plt.yticks() 부분 입니다.
수직 막대그래프는 plt.bar() 라고 코드를 입력해야 하지만 수평 막대그래프는 plt.barh() 라고 입력해야 합니다.
또한, 수직 막대그래프에서 x축 데이터의 값을 입력해주기 위해 plt.xticks() 를 사용했지만, 수평 막대그래프의 경우 y축에 해당 데이터 값을 입력해야 하기 때문에 plt.yticks() 를 사용했습니다.
아래 그래프는 똑같은 데이터로 수직 막대그래프를 그리는 코드로, 참고하시기 바랍니다 :D
import matplotlib.pyplot as plt
import numpy as np
x = np.arange(3)
item = ['coffee', 'tea', 'ade']
values = [2000, 4000, 5000]
plt.bar(x, values)
plt.xticks(x, item)
plt.show()