Plotting¶

import numpy as np
import matplotlib.pyplot as plt
from chebpy import chebfun

f = chebfun(lambda x: np.sin(10 * x) * np.exp(-x), [-1, 1])
f.plot()
plt.show()

fig, ax = plt.subplots()
f = chebfun(lambda x: np.sin(x), [0, 10])
g = chebfun(lambda x: np.cos(x), [0, 10])

f.plot(ax=ax, label="sin(x)")
g.plot(ax=ax, label="cos(x)")
ax.legend()
plt.show()

f = chebfun(lambda x: np.sin(x**2) + np.sin(x) ** 2, [0, 10])
g = chebfun(lambda x: np.exp(-((x - 5) ** 2) / 10), [0, 10])

ax = f.plot(label="f")
g.plot(ax=ax, label="g")

roots = (f - g).roots()
ax.plot(roots, f(roots), "ro", markersize=8, label="Intersections")
ax.legend()
ax.grid(True, alpha=0.3)
plt.show()