【備忘録】sympyによるグラフの図示

最近Pythonのモジュールであるnumpyやscipy, sympy, matplotlibを勉強中。
個人的備忘録として簡易的な関数の微分とグラフ図示を記事として纏める。

今回はガウス積分でよく出てくる
$$y = e ^ {-x^2} $$
これをサンプルとして使う。

python

import sympy as sp
import matplotlib.pyplot as plt

x = sp.symbols('x')
fx = sp.E ** -x ** 2
# f(x)の一階微分
fx_1 = sp.diff(fx, x)

p = sp.plot(fx, fx_1, (x, -3.5, 3), legend=True, show=False)
p[0].label = "$f(x) = e^{-x^2}$"
p[1].label = "$f(x) = -2xe^{-x^2}$"
p[1].line_color = "red"

plt.rcParams['figure.figsize'] = (9, 2.7)
p.show()