曲线拟合

猛男必看

import numpy as np
import matplotlib.pyplot as plt
from scipy import optimize
def f_1(x, A, B):
    return A*x + B
def f_2(x, A, B, C):
    return A*x*x + B*x + C
def f_3(x, A, B, C, D):
    return A*x*x*x + B*x*x + C*x + D
def plot_test():
    plt.figure()
    x0 = [1, 2, 3, 4, 5]
    y0 = [1, 3, 8, 18, 36]
    plt.scatter(x0[:], y0[:], 25, "red")
    A1, B1 = optimize.curve_fit(f_1, x0, y0)[0]
    x1 = np.arange(0, 6, 0.01)
    y1 = A1*x1 + B1
    plt.plot(x1, y1, "blue")

各位大神你们好，python菜鸟一个，想问下，程序倒数第四行最后的中括号里面写了个0是什么意思，还望不吝赐教

聂嘉辉

列表切片

猛男必看

聂嘉辉 发表于 2020-12-21 21:35
列表切片

就是要输出optimize.curve_fit函数根据x0,y0拟合出的A1和B1值对吗？