Computer >> 컴퓨터 >  >> 프로그래밍 >> Python

Python Matplotlib으로 사인 함수 그래프 그리는 방법

Matplotlib은 데이터 시각화를 위해 가장 널리 사용되는 Python 패키지 중 하나입니다. 데이터 시각화는 숫자를 일일이 살펴보거나 복잡한 계산을 수행하지 않고도 데이터 안에서 무슨 일이 일어나고 있는지 직관적으로 파악할 수 있게 해주는 핵심 과정입니다. 또한 정량적인 인사이트를 청중에게 효과적으로 전달하는 데 큰 도움이 됩니다.

Matplotlib은 데이터를 기반으로 2차원 플롯을 생성하는 데 사용됩니다. 객체 지향 API를 제공하여 생성한 플롯을 Python 애플리케이션에 손쉽게 내장할 수 있습니다. IPython 셸, Jupyter Notebook, Spyder IDE 등 다양한 환경에서 함께 사용할 수 있습니다.

Matplotlib은 Python으로 작성되었으며, 수치 연산 패키지인 NumPy를 기반으로 만들어졌습니다.

Matplotlib 설치하기

Windows 환경에서는 아래 명령어를 통해 Matplotlib을 간단히 설치할 수 있습니다.

pip install matplotlib

Matplotlib 설치 시 필요한 주요 의존성 패키지는 다음과 같습니다.

Python (버전 3.4 이상)
NumPy
Setuptools
Pyparsing
Libpng
Pytz
FreeType
Six
Cycler
Dateutil

사인 함수 그래프 그리기

이제 Matplotlib을 활용해 사인(sine) 함수를 플롯으로 나타내는 방법을 살펴보겠습니다.

예제 코드

import matplotlib.pyplot as plt
import numpy as np

data = np.arange(0.0, 4.0, 0.1)
y = 2 + np.sin(2 * np.pi * data)

fig, ax = plt.subplots()
ax.plot(data, y)
ax.set(xlabel='x-axis data', ylabel='y-axis data', title='A simple plot')
ax.grid()
plt.show()

출력 결과

Python Matplotlib으로 사인 함수 그래프 그리는 방법

코드 설명

  • 필요한 패키지들을 임포트하고, 사용 편의성을 위해 별칭(alias)을 정의합니다.

  • NumPy의 arange 함수를 사용해 0.0부터 4.0까지 0.1 간격의 x축 데이터를 생성합니다.

  • np.sin(2 * np.pi * data)를 통해 사인 값을 계산하고, 여기에 2를 더해 y축 데이터를 만듭니다.

  • plt.subplots() 함수를 사용해 figure 객체와 axes 객체를 동시에 생성합니다.

  • plot 함수를 사용해 x축 데이터와 y축 데이터를 플롯에 그립니다.

  • set 함수를 사용해 X축과 Y축의 레이블을 지정하고, 그래프의 제목을 설정합니다.

  • grid 함수를 호출해 그래프에 격자선을 추가하면 값을 읽기가 더욱 편해집니다.

  • 마지막으로 show 함수를 사용해 완성된 그래프를 화면에 출력합니다.

이처럼 Matplotlib을 활용하면 몇 줄의 코드만으로 사인 함수와 같은 수학적 함수를 손쉽게 시각화할 수 있습니다. NumPy와 결합하면 삼각함수뿐만 아니라 다양한 수학적 데이터를 직관적인 그래프로 표현할 수 있어, 데이터 분석 및 과학 컴퓨팅 분야에서 필수적인 도구로 자리 잡고 있습니다.