파이썬으로 통계 그래프 그리기: matplotlib 시작하기
파이썬에는 matplotlib라는 강력한 그래프 플로팅 라이브러리가 있습니다. matplotlib는 파이썬에서 가장 널리 사용되는 그래프 작성 및 데이터 시각화 모듈로, 단 세 줄의 코드만으로도 첫 번째 그래프를 그릴 수 있습니다.
가장 간단한 그래프 예제
from matplotlib import pyplot as plt # 캔버스에 그래프 그리기 plt.plot([1,2,3],[4,5,1]) # 그린 그래프를 화면에 표시하기 plt.show()
위 코드를 실행하면 좌표 (1,4), (2,5), (3,1)을 연결하는 간단한 선 그래프가 생성됩니다.
축 레이블과 제목 추가하기
xlabel과 ylabel 함수를 사용하면 각 축에 레이블을 지정할 수 있습니다.
plt.ylabel('Y축')
plt.xlabel('X축')
또한 title 함수를 사용해 그래프에 제목을 붙일 수 있습니다.
plt.title('내 그래프')
주의: 이 함수들은 반드시 show() 호출 이전에 실행해야 합니다. show()가 먼저 호출되면 이후에 설정한 레이블이나 제목은 화면에 반영되지 않습니다.
그래프 스타일 꾸미기
plt.plot 함수에 linewidth=2 같은 인자를 전달하면 선 두께 등 그래프의 스타일을 자유롭게 조절할 수 있습니다. 또한 legend() 함수를 호출하면 그래프에 범례를 추가할 수 있습니다.
plt.legend()
다양한 차트 종류 활용하기
matplotlib는 막대그래프(bar graph), 산점도(scatter plot), 원형 차트(pie chart) 등 다양한 종류의 차트를 지원합니다. 각 차트 유형별 사용법을 더 자세히 알고 싶다면 matplotlib 공식 튜토리얼을 참고해 보세요.