如何使用Python轻松地检入非线性关系？

import numpy as np
import matplotlib.pyplot as plt
from scipy.optimize import curve_fit

def nonlinear_func(x, a, b, c):
    return a * np.exp(-b * x) + c

x_data = np.array([1, 2, 3, 4, 5])
y_data = np.array([2.5, 1.6, 1.1, 0.6, 0.3])

popt, pcov = curve_fit(nonlinear_func, x_data, y_data)

x_fit = np.linspace(1, 5, 100)
y_fit = nonlinear_func(x_fit, *popt)

plt.plot(x_data, y_data, 'bo', label='data')
plt.plot(x_fit, y_fit, 'r-', label='fit')
plt.legend()
plt.xlabel('x')
plt.ylabel('y')
plt.show()