Computer >> 컴퓨터 >  >> 프로그래밍 >> Python

파이썬 Matplotlib로 꺾은선 그래프(Line Chart) 만드는 방법

데이터 시각화에서 가장 기본이 되는 그래프 중 하나가 바로 꺾은선 그래프(Line Chart)입니다. 파이썬의 대표적인 시각화 라이브러리인 Matplotlib를 사용하면 몇 줄의 코드만으로 간단하게 꺾은선 그래프를 그릴 수 있습니다.

꺾은선 그래프 만들기 단계

  • 그림(figure) 크기를 설정하고, 서브플롯(subplot) 주변과 사이의 여백(padding)을 조정합니다.
  • 연도(years)와 인구 증가(population) 데이터를 리스트로 준비합니다.
  • plot() 메서드를 사용해 연도와 인구 데이터를 꺾은선으로 그립니다.
  • show() 메서드를 호출하여 그래프를 화면에 표시합니다.

예제 코드

from matplotlib import pyplot as plt

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

years = [1901, 1911, 1921, 1931, 1941, 1951,
         1961, 1971, 1981, 1991, 2001, 2011]
population = [237.4, 238.4, 252.09, 251.31, 278.98,
              318.66, 361.09, 439.23, 548.16,
              683.33, 846.42, 1028.74]

plt.plot(years, population, color='red', marker='o')

plt.show()

코드 설명

plt.rcParams를 이용하면 그래프의 전역 설정을 변경할 수 있습니다. 위 예제에서는 figure.figsize로 그림 크기를 가로 7.50인치, 세로 3.50인치로 지정했고, figure.autolayoutTrue로 설정하여 레이블이나 제목이 잘리지 않도록 여백을 자동 조정했습니다.

plot() 메서드에는 다양한 스타일 옵션을 적용할 수 있습니다. 이 예제에서는 color='red'로 선 색상을 빨간색으로 지정하고, marker='o'를 추가해 각 데이터 포인트를 원형 마커로 표시했습니다. 이 외에도 linestyle(선 스타일), linewidth(선 두께) 등의 옵션을 활용하면 그래프를 더욱 다양하게 꾸밀 수 있습니다.

실행 결과

위 코드를 실행하면 x축에는 연도(1901~2011), y축에는 인구 수가 표시되며, 빨간색 선과 원형 마커로 연도별 인구 증가 추이를 한눈에 확인할 수 있는 꺾은선 그래프가 출력됩니다.