더북(TheBook)

8.2.1 막대그래프 그리기

막대그래프는 꺾은선 그래프만큼 자주 쓰입니다. 응용할 수 있는 부분도 무궁무진하죠. 막대그래프는 plt.bar()로 그립니다. 꺾은선 그래프를 그린 것처럼 다음 코드로 막대그래프를 그려 봅시다.

import matplotlib.pyplot as plt

sales = [1, 5, 7, 3, 7]
plt.bar(sales)
plt.show()
실행결과
----> 4 plt.bar(sales)
TypeError: bar() missing 1 required positional argument: 'height'

오류가 뜨지만 당황하지 말고 오류 내용을 잘 살펴봅시다. 마지막 줄을 보니 TypeError가 발생했고, ‘bar()에 요구되는 위치 요소인 높이(height)가 누락됨’이라고 합니다. 즉, bar()에는 높이도 넣어 줘야 하는데, 빠뜨렸다는 뜻입니다.

plt.plot()은 데이터를 하나만 넣어도 됐지만, plt.bar()는 2개의 데이터(x축, y축)를 모두 넣어야 그래프를 그립니다. 그럼 데이터를 추가해 다시 실행해 봅시다.

sales = [1, 5, 7, 3, 7]
month = [3, 4, 5, 6, 7]
plt.bar(month, sales)
plt.show()
실행결과

막대그래프가 잘 그려집니다. x축에 month의 값이 차례대로 들어가고, y축에도 sales의 값이 차례대로 들어갑니다.

신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.