Python画出美观的论文图（5） 零点求解与基于diff的导函数绘制

from scipy.special import airy
from scipy import optimize

def f(x):
    return airy(-x)[0]

x = np.linspace(-3, 8, 201)
y = f(x)
zeros = [optimize.newton(f,x) for x in x[np.nonzero(np.diff(np.sign(y)))]]
plt.axvline(0,color='black',lw=1)
plt.axhline(0,color='black',lw=1)
plt.scatter(zeros,np.zeros_like(zeros),c='r')
plt.plot(x,y)
plt.plot(x[:-1],200/11*np.diff(y),'--')