红球一共6组，每组从1-33中抽取一个，六个互相不重复。然后蓝球是从1-16中抽取一个数字，这整个组成的双色球
forindexinresult:dit={'期号':index['code'],'开奖日期':index['date'],'红球':index['red'],'蓝球':index['blue'],'一等奖中奖注数':index['prizegrades'][0]['typenum'],'一等奖中奖金额':index['prizegrades'][0]['typemoney'],'二等奖中奖注数':index['prizegrades'][1]['typenum'],'二等奖中奖金额':index['prizegrades'][1]['typemoney'],'三等奖中奖注数':index['prizegrades'][2]['typenum'],'三等奖中奖金额':index['prizegrades'][2]['typemoney'],'四等奖中奖注数':index['prizegrades'][3]['typenum'],'四等奖中奖金额':index['prizegrades'][3]['typemoney'],'五等奖中奖注数':index['prizegrades'][4]['typenum'],'五等奖中奖金额':index['prizegrades'][4]['typemoney'],'六等奖中奖注数':index['prizegrades'][5]['typenum'],'六等奖中奖金额':index['prizegrades'][5]['typemoney'],'一等奖中奖地区':index['content'],'奖池金额':index['poolmoney']}
importcsv#内置模块f=open('双色球.csv',mode='a',encoding='utf-8',newline='')csv_writer=csv.DictWriter(f,fieldnames=['期号','开奖日期','红球','蓝球','一等奖中奖注数','一等奖中奖金额','二等奖中奖注数','二等奖中奖金额','三等奖中奖注数','三等奖中奖金额','四等奖中奖注数','四等奖中奖金额','五等奖中奖注数','五等奖中奖金额','六等奖中奖注数','六等奖中奖金额','一等奖中奖地区','奖池金额'])csv_writer.writeheader()#写入表头csv_writer.writerow(dit)print(dit)
运行代码，这样就得到了往期双色球的数据了
importpandasaspdfrompyecharts.chartsimport*fromsklearn.linear_modelimportLogisticRegression
data=pd.read_csv('双色球.csv',encoding='utf-8',engine='python')data.head()
defget_lotto_data(data,lotto,lotto_id):data['lotto_id']=lotto_idX=[]Y=[]#标签and值fors,pinzip(data['lotto_id'],data[lotto]):X.append([float(s)])Y.append(float(p))returnX,Y
deflinear_model_test(X,Y,predict_value):regr=LogisticRegression()regr.fit(X,Y)predict_outcome=regr.predict(predict_value)predictions={}predictions['intercept']=regr.intercept_predictions['coefficient']=regr.coef_predictions['predicted_value']=predict_outcomereturnpredictions
defget_predicted_num(file,lotto,lotto_id):X,Y=get_lotto_data(file,lotto,lotto_id)predict_value=[[33]]result=linear_model_test(X,Y,predict_value)iflotto_id<7:print(f'中奖第{lotto_id}个红球为：',result['predicted_value'].astype('int64'),'号球')else:print('中奖蓝球为：',result['predicted_value'].astype('int64'),'号球')
get_predicted_num(data,'r1',1)#预测红1get_predicted_num(data,'r2',2)#预测红2get_predicted_num(data,'r3',3)#预测红3get_predicted_num(data,'r4',4)#预测红4get_predicted_num(data,'r5',5)#预测红5get_predicted_num(data,'r6',6)#预测红6get_predicted_num(data,'蓝球',7)#预测蓝7
x=red_ball_count.index.tolist()y=red_ball_count.values.tolist()#可视化展示pie=(Pie().add("",[list(z)forzinzip(x,y)]))pie.render_notebook()
x=blue_ball_count.index.tolist()y=blue_ball_count.values.tolist()pie=(Pie().add("",[list(z)forzinzip(x,y)]))pie.render_notebook()
frompyechartsimportoptionsasoptsfrompyecharts.chartsimportPictorialBarfrompyecharts.globalsimportSymbolTypec=(PictorialBar().add_xaxis(x).add_yaxis("",y,label_opts=opts.LabelOpts(is_show=False),symbol_size=18,symbol_repeat="fixed",symbol_offset=[0,0],is_symbol_clip=True,).reversal_axis().set_global_opts(title_opts=opts.TitleOpts(title='蓝球中奖号码'),xaxis_opts=opts.AxisOpts(is_show=False),yaxis_opts=opts.AxisOpts(axistick_opts=opts.AxisTickOpts(is_show=False),axisline_opts=opts.AxisLineOpts(linestyle_opts=opts.LineStyleOpts(opacity=0)),),))c.render_notebook()
x_data=df['中奖注数'].index.tolist()y_data=df['中奖注数'].values.tolist()c=(Funnel().add("中奖注数漏斗图",[list(z)forzinzip(x_data,y_data)],label_opts=opts.LabelOpts(position="inside"),).set_global_opts(title_opts=opts.TitleOpts(title="中奖注数漏斗图")))c.render_notebook()