Python数据分析之双色球基于线性回归算法预测

@本文来源于公众号：csdn2299，喜欢可以关注公众号 程序员学府
本文实例讲述了Python数据分析之双色球基于线性回归算法预测下期中奖结果。分享给大家供大家参考，具体如下：

前面讲述了关于双色球的各种算法，这里将进行下期双色球号码的预测，想想有些小激动啊。

代码中使用了线性回归算法，这个场景使用这个算法，预测效果一般，各位可以考虑使用其他算法尝试结果。

发现之前有很多代码都是重复的工作，为了让代码看的更优雅，定义了函数，去调用，顿时高大上了

    #!/usr/bin/python
    # -*- coding:UTF-8 -*-
    #导入需要的包
    import pandas as pd
    import numpy as np
    import matplotlib.pyplot as plt
    import operator
    from sklearn import datasets,linear_model
    from sklearn.linear_model import LogisticRegression
    #读取文件
    df = pd.read_table('newdata.txt',header=None,sep=',')
    #读取日期
    tdate = sorted(df.loc[:,0])
    #将以列项为数据，将球号码取出，写入到csv文件中，并取50行数据
    # Function to red number to csv file
    def RedToCsv(h_num,num,csv_name):
     h_num = df.loc[:,num:num].values
     h_num = h_num[50::-1]
     renum2 = pd.DataFrame(h_num)
     renum2.to_csv(csv_name,header=None)
     fp = file(csv_name)
     s = fp.read()
     fp.close()
     a = s.split('\n')
     a.insert(0, 'numid,number')
     s = '\n'.join(a)
     fp = file(csv_name, 'w')
     fp.write(s)
     fp.close()
    #调用取号码函数
    # create file
    RedToCsv('red1',1,'rednum1data.csv')
    RedToCsv('red2',2,'rednum2da