python : 蒙特卡罗算法 应用于双色球

参考书：算法设计与分析 王晓东 编著 ：第7章 概率算法 7.5 蒙特卡罗算法 

http://www.gdfc.org.cn/datas/history/twocolorball/history_1.html 抓取双色球开奖数据

2017001,09,11,14,20,25,26,15
2017002,15,19,23,24,25,32,03
2017003,01,04,08,15,27,32,16

 ... ...

存为数据文件: cp2017.txt

自设seed种子，随机生成1注6个红球，计算这1组数字与双色球开奖历史数据之相似性。

redball.py

# -*- coding: cp936 -*-
import os, sys
import random

if len( sys.argv ) ==2:
    p1 = long(sys.argv[1])
else:
    print 'usage: redball.py seed_int '
    sys.exit(1)

f1 = "cp2017.txt"
if not os.path.exists(f1):
    print 'ERROR: %s is not found.' % f1
    sys.exit(1)

# 计算红球相似性
def mc(A , B):
    k=0
    for a in A:
        if a in B:
            k +=1
    return k
#

A = [ 0 for i in range(0,7)] # 初始化一个具有6个0的数组,
# 随机选出6个红球
random.seed(p1)
reds = [] 
while len(reds) < 6:
    N = random.randint(1,33)
    if N not in reds:
        reds.append(N)
print 'random:',sorted(reds)

fp = open(f1,'r')        
alist =[] 
ln =0
for line in fp:
    alist = line.strip().split(',')
    for i in range(1,7):
        A[i] = int(alist[i])
    k = mc(reds, A[1:])
    if k > 3:
        ln += 1
        print line.rstrip(),':',k
#
fp.close()
print 'ln=',ln

例如: 执行 redball.py  1254436

random: [4, 10, 11, 25, 30, 31]

2017115,04,10,11,25,30,31,01 : 6