[파이썬] 꺾은선 그래프(스텝 플롯) 그리기

꺾은선 그래프(스텝 플롯)는 데이터 포인트를 선으로 연결하여 시계열 데이터나 변화 과정을 시각화하는데 사용되는 효과적인 도구입니다. Python의 matplotlib 라이브러리를 사용하여 꺾은선 그래프를 그릴 수 있습니다. 이번 포스트에서는 꺾은선 그래프를 그리기 위한 간단한 예제 코드를 소개하겠습니다.

필요한 라이브러리 설치

먼저, matplotlib 라이브러리를 설치해야 합니다. 아래의 명령어를 사용하여 라이브러리를 설치하세요.

pip install matplotlib

꺾은선 그래프 그리기

다음은 꺾은선 그래프를 그리기 위한 간단한 예제 코드입니다. 코드를 한 줄씩 이해해보세요.

import matplotlib.pyplot as plt

# X축에 해당하는 데이터
x = [1, 2, 3, 4, 5]

# Y축에 해당하는 데이터
y = [2, 4, 6, 8, 10]

# 꺾은선 그래프 그리기
plt.step(x, y)

# 그래프 제목 설정
plt.title("Step Plot")

# X축 레이블 설정
plt.xlabel("X-axis")

# Y축 레이블 설정
plt.ylabel("Y-axis")

# 그래프 출력
plt.show()

위 코드를 실행하면 다음과 같은 꺾은선 그래프가 생성됩니다.

step_plot

그래프 스타일 설정

matplotlibplot 함수와 마찬가지로, step 함수로 그린 꺾은선 그래프의 스타일을 다양하게 설정할 수 있습니다. 예를 들면, 선의 색깔, 선의 종류, 두께 등을 조절할 수 있습니다.

import matplotlib.pyplot as plt

x = [1, 2, 3, 4, 5]
y = [2, 4, 6, 8, 10]

plt.step(x, y, color='red', linestyle='--', linewidth=2)

plt.title("Step Plot")
plt.xlabel("X-axis")
plt.ylabel("Y-axis")

plt.show()

위 코드는 꺾은선 그래프의 선 색깔을 빨간색으로, 선의 종류를 점선으로, 두께를 2로 설정한 예제입니다. 원하는 스타일에 맞게 step 함수의 인자를 바꾸어보세요.

마무리

위 예제 코드를 통해 파이썬의 matplotlib 라이브러리를 사용하여 꺾은선 그래프를 그리는 방법을 알아보았습니다. 꺾은선 그래프는 시계열 데이터나 변화 과정 시각화에 매우 유용한 도구입니다. matplotlib 라이브러리의 다른 함수들을 활용하면 다양한 스타일의 꺾은선 그래프를 그릴 수 있습니다.