python实现蓄水池抽样问题

</pre><pre name="code" class="python">import random
filename='d:/test.txt'//要抽样的文件
f=open(filename,'r')
sampleNum=1000//样本个数
samples=[]//样本列表

currentLineNum=sampleNum

for i in range(sampleNum):
samples.append(f.readline())

while True:
currentLine=f.readline()
if currentLine=='':
break
currentLineNum=currentLineNum+1
r=random.randint(1,currentLineNum)
if r<sampleNum:
samples[r-1]=currentLine