第十六天学习进度--数列找规律算法（预测算法）之更一般形式（2）

import numpy as np
from scipy.optimize import curve_fit
import scipy.stats as stats

def func_exxaxx(x,a,b,c,d,e,f,g):
    x=np.array(x)
    y = a*np.exp(b*x)+c*x**d+e*x**(f*x)+g
    return y


def func_exxa(x,a,b,c,d,g):
    x=np.array(x)
    y = a*np.exp(b*x)+c*x**d+g
    return y


def func_exxx(x,a,b,e,f,g):
    x=np.array(x)
    y = a*np.exp(b*x)+e*x**(f*x)+g
    return y


def func_xaxx(x,c,d,e,f,g):
    x=np.array(x)
    y = c*x**d+e*x**(f*x)+g
    return y

def func_ex(x,a,b,g):
    x=np.array(x)
    y = a*np.exp(b*x)+g
    return y

def func_xa(x,c,d,g):
    x=np.array(x)
    y = c*x**d+g
    return y

def func_xx(x,e,f,g):
    x=np.array(x)
    y =e*x**(f*x)+g
    return y

allfunc=[func_exxaxx,func_exxa,func_exxx,func_xaxx,func_ex,func_xa,func_xx]

def polyfit(x, y,limitfunc=lambda x:x**3):
    fit_coef_list, pcov_list=[],[]
    maxnh=0
    id=0
    suit=0
    suiti=0
    for i,one in enumerate(allfunc):
        try:
            fit_coef, pcov = curve_fit(one, x, y,maxfev=1500000)
            fit_coef_list.append(fit_coef)
            pcov_list.append(pcov)
            result=one([x],*fit_coef)[0]
            finalone=one([len(x)+1],*fit_coef)[0]
            downs=0.5 if abs(finalone-result[-1])>limitfunc(max(result)-min(result)) else 0
            a,b=stats.pearsonr(result, y)
            score=a*(1-b)-downs
            print(one,score)
            id+=1
            if(score>maxnh):
                maxnh=score
                suit=id-1
                suiti=i
        except Exception:
            pass
    print(allfunc[suiti])
    #参数
    return fit_coef_list[suit],allfunc[suiti]

def find_logical(y,limitfunc=lambda x:x**3):
    x_None=[]
    x=[]
    y_temp=[]
    for i,info in enumerate(y):
        if(info!=None):
            x.append(i+1)
            y_temp.append(info)
        else:
            x_None.append(i)
    fit_coef,func = polyfit(x, y_temp,limitfunc)
    for one in x_None:
        x.insert(one,None)
    return fit_coef,x,func

#函数对应

y=[4,21,64,145,276]
x=list(range(1,len(y)+1,1))
fit_coef,func = polyfit(x, y)
print(func([6],*fit_coef))

#数列规律(数学归纳法找第六位是什么)
fit_coef,x,func=find_logical([11,111,1111,11111,111111])
print(func([6],*fit_coef))

#数列规律寻找下一项
fit_coef,x,func=find_logical([1,2,3,5,8,13,21,34,55,89,144,233,377,610,987,1597,2584,4181,6765])
print(func([len(x)+1],*fit_coef))