드디어 그래프가 그려집니다. 코드와 결과를 살펴봅시다. plot()에 넣은 리스트 [1, 5, 7, 3, 7]은 x축일까요? y축일까요? 그래프를 보면 y축으로 보이죠? 이처럼 plt.plot()은 데이터가 단일 값이면 입력된 값을 y축으로 보고 x축을 기본으로 0부터 설정됩니다. 만약 x축에도 특정 값을 넣고 싶다면 x축 데이터도 추가하면 됩니다. 2개의 데이터를 넣으면 앞은 x축, 뒤는 y축으로 인식해 그래프를 그립니다.

형식

plt.plot(x축, y축)

그래프를 그리려면 앞에서처럼 여러 개의 데이터 즉, 데이터 묶음이 필요합니다. 이 책에서는 리스트로 그래프를 그리겠습니다.

앞에서는 plt.plot()에 리스트를 직접 넣었지만, 리스트를 변수에 저장하고 변수를 넣는 것이 일반적입니다. [1, 5, 7, 3, 7]을 3월부터 7월까지 5개월 간의 매출이라고 가정해 봅시다. 그럼 3월부터 7월을 리스트 [3, 4, 5, 6, 7]로 만들고, 이를 변수 month에 저장합니다. [1, 5, 7, 3, 7]도 변수 sales에 저장합니다. month와 sales를 plt.plot()에 넣으면 각각 x축과 y축을 나타내는 리스트가 되겠죠? 다음 코드를 실행해 봅시다.

month = [3, 4, 5, 6, 7]
sales = [1, 5, 7, 3, 7]
plt.plot(month, sales)
plt.show()

실행결과