파이썬에서 배열 데이터를 시각화하는 가장 기본적인 방법은 Matplotlib 라이브러리를 활용하는 것입니다. 이 글에서는 NumPy 배열을 생성하고 이를 선 그래프로 그리는 과정을 단계별로 살펴보겠습니다.
배열 그래프를 그리는 단계
- 먼저 그림(figure)의 크기를 설정하고, 서브플롯 주변과 사이의 여백(padding)을 조정합니다.
- NumPy를 사용하여 두 개의 배열 x와 y를 생성합니다.
- title() 메서드를 사용해 그래프의 제목을 지정합니다.
- x와 y 데이터 포인트를 빨간색 선으로 플롯합니다.
- 마지막으로 show() 메서드를 호출하여 그림을 화면에 표시합니다.
예제 코드
import numpy as np
import matplotlib.pyplot as plt
plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True
x = np.array([5, 4, 1, 4, 5])
y = np.sort(x)
plt.title("Line graph")
plt.plot(x, y, color="red")
plt.show()실행 결과
코드 설명
위 예제에서 plt.rcParams["figure.figsize"]는 그림의 가로·세로 크기를 인치 단위로 지정하며, figure.autolayout을 True로 설정하면 레이블이나 제목이 잘리지 않도록 여백이 자동으로 조절됩니다.
배열 x는 임의의 값 5개로 구성되어 있고, y는 np.sort() 함수를 통해 오름차순으로 정렬된 값입니다. 따라서 두 배열을 선 그래프로 연결하면 데이터가 정렬되는 과정을 한눈에 확인할 수 있습니다.
plt.plot(x, y, color="red")에서 color 인자를 변경하면 파란색(blue), 녹색(green) 등 원하는 색상으로 그래프를 표현할 수 있으며, 마커나 선 스타일 옵션을 추가하면 더욱 다양한 시각화가 가능합니다.