[python] SymPy를 사용하여 수학 함수 그리기

이번 포스트에서는 Python의 SymPy 라이브러리를 사용하여 수학 함수를 그리는 방법을 알아보겠습니다.

SymPy는 Python의 대표적인 기호 연산 라이브러리로, 수학 함수의 기호적인 연산과 그래프 그리기를 지원합니다. 이를 통해 수학적인 문제를 해결하거나 수학 함수를 시각화할 수 있습니다.

SymPy 설치

먼저 SymPy를 설치해야 합니다. 아래의 명령어를 사용하여 SymPy를 설치할 수 있습니다.

pip install sympy

수학 함수 그리기

이제 SymPy를 사용하여 수학 함수를 그려보겠습니다. 먼저 SymPy와 그래프 플로팅을 위한 Matplotlib 라이브러리를 import합니다.

import sympy as sp
import matplotlib.pyplot as plt

그리고 심볼 변수를 생성하여 수학 함수에 사용합니다.

x = sp.symbols('x')

이제 그래프를 그릴 함수를 정의합니다. 예를 들어, x^2 함수를 그리기 위해서는 다음과 같이 작성할 수 있습니다.

func = x**2

그리고 x의 범위를 지정합니다.

x_range = sp.Interval(-5, 5)

이제 sympy.plot 함수를 사용하여 실제로 그래프를 그립니다.

sp.plot(func, (x, x_range.start, x_range.end))
plt.show()

위의 코드를 실행하면 x^2 함수의 그래프가 표시됩니다.

추가적인 설정

그래프를 좀 더 멋지게 꾸미고 싶다면, Matplotlib의 설정을 사용할 수 있습니다. 예를 들어, 그래프의 색상, 라인 스타일, 레이블 등을 변경할 수 있습니다.

예를 들어, 그래프의 라인 색상을 변경하고 레이블을 추가하는 방법은 다음과 같습니다.

sp.plot(func, (x, x_range.start, x_range.end), line_color='red', label='y = x^2')
plt.legend()
plt.show()

위의 코드를 실행하면 빨간색 선으로 표시되는 x^2 함수의 그래프가 표시되고, 그래프에 “y = x^2”라는 레이블이 표시됩니다.

마무리

이번 포스트에서는 SymPy를 사용하여 수학 함수를 그리는 방법에 대해 알아보았습니다. SymPy의 다양한 기능을 활용하면 더 복잡한 수학 함수도 그릴 수 있습니다. SymPy의 자세한 사용법은 공식 문서를 참고하시기 바랍니다.

다음 포스트에서는 SymPy를 사용하여 미적분, 행렬 연산 등을 수행하는 방법에 대해 알아보도록 하겠습니다.