PyCkBk-4-6

文章来自《Python cookbook》.

翻译仅仅是为了个人学习,其它商业版权纠纷与此无关!

-- 61.182.251.99 [2004-09-21 05:42:12]

描述

Retrieving a Line at Random from a File of Unknown Size

读取未知大小文件的随机一行

问题 Problem

有文件,不清楚大小(但是可能非常大), 需要忽略文件, 只读取数据的随机一行。

解决 Solution

We do need to read the whole file, but we don't have to read it all at once:

确实需要读取文件的全部数据, 但不是一次全部读出:

   1 import random
   2 
   3 def randomLine(file_object):
   4     "顺序读取文件内容,取文件的随机一行"
   5     lineNum = 0
   6     selected_line = ''
   7 
   8     while 1:
   9         aLine = file_object.readline(  )
  10         if not aLine: break
  11         lineNum = lineNum + 1
  12         # 是否选择本行?                      #译注:这里英语原文注释可能有错,根据程序逻辑译出   
  13         if random.uniform(0,lineNum)<1:    #译注: 算法的解释见讨论部分,请先考虑考虑,uniform函数用法附后            
  14             selected_line = aLine          
  15     file_object.close(  )
  16     return selected_line

讨论 Discussion

当然, 更直接的方法是这样的:

random.choice(file_object.readlines(  ))

但是, 这需要将全部文件内容读入内存, 对确实很大的文件可能有问题。

解决部分中算法的理论依据不是很明显,但是也不是太深奥: 当读取到文件第行时,当前行(即变量selected_line指向的行)被随机取得的可能性是1/N。 容易看出算法选择当前最后读取的一行的可能性是对的,因为此处我们明确地以1.0/lineNum可能性选择最后一行。

由归纳法可以证明算法的合理性:如果读取第N-1行时,选择N-1行的可能性是对的,那么显然对于第N行时算法也是对的。由于(仅读取到第一行)选择第一行的可能性是1, 因此,算法的随机选择处理直到第N行都是对的。

同样的技巧对于从有限序列中,应用均匀分布的可能性,一次处理一个元素(原因未知)地随机选择一个元素也是适用的。但是除了从比如/usr/dict/words 这样的文件中随机选择一个单词以外,这个理论的应用不是很广。

参考 See Also

Python 库参考 random模块部分

#译注:为阅读方便,这里给出random模块两个函数的说明

uniform( a, b) :Return a random real number N such that a <= N < b.

choice( seq) :Return a random element from the non-empty sequence seq.

PyCkBk-4-6 (last edited 2009-12-25 07:16:21 by localhost)