파이썬(Python) 그래프 그리기(matplotlib)

Python으로 그래프를 그리려면 그래프를 그릴 수 있는 라이브러리가 필요합니다. Python에는 많은 그래프 그리기 라이브러리가 있습니다. 예를 들어, Matplotlib, Seaborn, Plotly, Bokeh 등이 있습니다.

그래프를 그리기 위해서는 우선 해당 라이브러리를 설치해야 합니다. 아래와 같은 방법으로 설치할 수 있습니다.

pip install matplotlib

 

이후, Python 코드에서 Matplotlib을 사용하기 위해서는 아래와 같이 import 해야 합니다.

import matplotlib.pyplot as plt

 

그래프를 그리는 기본적인 절차는 다음과 같습니다.

1. 그래프를 그리기 위한 데이터를 준비합니다.
2. 그래프를 그리기 위한 Figure 객체를 생성합니다.
3. Figure 객체 위에 Axes 객체를 생성합니다.
4. Axes 객체에 그래프를 그리기 위한 메소드(ex: plot, scatter, bar)를 사용합니다.
5. 그래프를 출력합니다.

 

예를 들어, 선 그래프를 그리기 위한 코드는 다음과 같습니다.

import matplotlib.pyplot as plt

# 그래프를 그리기 위한 데이터
x = [1, 2, 3, 4]
y = [1, 4, 9, 16]

# Figure 객체 생성
fig = plt.figure()

# Figure 객체 위에 Axes 객체 생성
ax = fig.add_subplot()

# Axes 객체에 선 그래프 그리기
ax.plot(x, y)

# 그래프 출력
plt.show()

만약, 산점도 그래프(scatter plot)를 그리고 싶다면 아래와 같이 코드를 작성할 수 있습니다.

import matplotlib.pyplot as plt

# 그래프를 그리기 위한 데이터
x = [1, 2, 3, 4]
y = [1, 4, 9, 16]

# Figure 객체 생성
fig = plt.figure()

# Figure 객체 위에 Axes 객체 생성
ax = fig.add_subplot()

# Axes 객체에 산점도 그래프 그리기
ax.scatter(x, y)

# 그래프 출력
plt.show()

그래프의 제목과 축 이름을 추가하고 싶다면, 아래와 같이 그래프 그리기 코드에 추가할 수 있습니다.

import matplotlib.pyplot as plt

# 그래프를 그리기 위한 데이터
x = [1, 2, 3, 4]
y = [1, 4, 9, 16]

# Figure 객체 생성
fig = plt.figure()

# Figure 객체 위에 Axes 객체 생성
ax = fig.add_subplot()

# Axes 객체에 선 그래프 그리기
ax.plot(x, y)

# 그래프 제목 추가
ax.set_title("My Graph")

# x 축 이름 추가
ax.set_xlabel("X-axis")

# y 축 이름 추가
ax.set_ylabel("Y-axis")

# 그래프 출력
plt.show()

위 코드는 "My Graph"라는 제목과 "X-axis", "Y-axis"라는 축 이름이 추가된 선 그래프가 출력됩니다.

 

그래프의 색상과 마커 스타일도 추가할 수 있습니다.

import matplotlib.pyplot as plt

# 그래프를 그리기 위한 데이터
x = [1, 2, 3, 4]
y = [1, 4, 9, 16]

# Figure 객체 생성
fig = plt.figure()

# Figure 객체 위에 Axes 객체 생성
ax = fig.add_subplot()

# Axes 객체에 선 그래프 그리기
ax.plot(x, y, color='red', marker='o')

# 그래프 제목 추가
ax.set_title("My Graph")

# x 축 이름 추가
ax.set_xlabel("X-axis")

# y 축 이름 추가
ax.set_ylabel("Y-axis")

# 그래프 출력
plt.show()

 

위 코드는 "o"라는 마커 스타일과 "red"라는 색상을 갖는 선 그래프가 출력됩니다.

 

그래프의 범례(legend)도 추가할 수 있습니다.

import matplotlib.pyplot as plt

# 그래프를 그리기 위한 데이터
x = [1, 2, 3, 4]
y1 = [1, 4, 9, 16]
y2 = [2, 8, 18, 32]

# Figure 객체 생성
fig = plt.figure()

# Figure 객체 위에 Axes 객체 생성
ax = fig.add_subplot()

# Axes 객체에 선 그래프 그리기
ax.plot(x, y1, label='y1')
ax.plot(x, y2, label='y2')

# 그래프 제목 추가
ax.set_title("My Graph")

# x 축 이름 추가
ax.set_xlabel("X-axis")

# y 축 이름 추가
ax.set_ylabel("Y-axis")

# 범례 추가
ax.legend()

# 그래프 출력
plt.show()

위 코드는 "y1"과 "y2"라는 범례가 추가된 선 그래프가 출력됩니다.