Matplotlib로 데이터 시각화하기
데이터를 분석할 때 숫자로 된 데이터만 보는 것보다 그래프를 활용하면 훨씬 직관적으로 이해할 수 있습니다.
Matplotlib(맷플롯립)은 파이썬에서 데이터를 그래프로 시각화하는 가장 기본적인 라이브러리입니다.
이번 강의에서는 Matplotlib의 기본 사용법과 선 그래프, 막대 그래프를 그리는 방법을 배워보겠습니다.
Matplotlib 설치하기
Matplotlib은 다음 명령어로 설치할 수 있습니다. 실습 환경에는 Matplotlib이 이미 설치되어 있어 별도로 설치할 필요는 없습니다.
pip install matplotlib
1. Matplotlib 설치 및 기본 사용법
아래 코드를 실행하면 x값을 가로축, y값을 세로축으로 하는 선 그래프가 출력됩니다.
import matplotlib.pyplot as plt
x = [1, 2, 3, 4, 5]
y = [10, 20, 25, 30, 40]
plt.plot(x, y)
plt.title("기본 선 그래프")
plt.xlabel("X 축")
plt.ylabel("Y 축")
plt.show()
plt.plot(x, y)는 x축과 y축 데이터를 사용하여 선 그래프를 그리는 함수입니다.
plt.title(), plt.xlabel(), plt.ylabel()는 그래프 제목과 축 이름을 설정합니다.
plt.show()는 그래프를 화면에 표시합니다.
2. 그래프 스타일 설정하기
Matplotlib에서는 그래프의 색상, 선 스타일, 마커(marker) 등을 조정할 수 있습니다.
import matplotlib.pyplot as plt
x = [1, 2, 3, 4, 5]
y = [10, 20, 25, 30, 40]
plt.plot(x, y, color='red', linestyle='--', marker='o')
plt.title("스타일이 적용된 선 그래프")
plt.xlabel("X 축")
plt.ylabel("Y 축")
plt.show()
📌 주요 옵션
-
color='red': 선 색상을 빨간색으로 설정 -
linestyle='--': 점선 스타일 적용 -
marker='o': 데이터 포인트에 원형(o) 마커 추가
이러한 스타일 설정을 활용하면 더 직관적인 그래프를 만들 수 있습니다.
3. 막대 그래프 그리기
Matplotlib을 사용하면 카테고리별 데이터를 쉽게 비교할 수 있는 막대 그래프(bar chart) 를 그릴 수 있습니다.
import matplotlib.pyplot as plt
labels = ["A", "B", "C", "D"]
values = [30, 70, 50, 90]
plt.bar(labels, values, color=['red', 'blue', 'green', 'orange'])
plt.title("기본 막대 그래프")
plt.xlabel("카테고리")
plt.ylabel("값")
plt.show()
plt.bar(x, y)는 카테고리와 값을 입력하여 막대 그래프를 그리는 함수입니다.
color=['red', 'blue', 'green', 'orange']는 막대의 색상을 각각 지정합니다.
이러한 막대 그래프는 카테고리별 데이터를 비교할 때 유용합니다.
다음 강의에서는 히스토그램, 산점도, 파이 차트, 그리고 서브플롯(subplot)을 다뤄보겠습니다.
다음 내용이 궁금하다면?
코 드프렌즈 PLUS 멤버십 가입 or 강의를 등록해 주세요!