文章来自《Python cookbook》. 翻译仅仅是为了个人学习,其它商业版权纠纷与此无关!
-- 61.182.251.99 [DateTime(2004-09-21T05:42:12Z)] TableOfContents
描述
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:
这里确实需要读取文件的全部数据,但不是一次全部读出:
import random def randomLine(file_object): "顺序读取文件内容,取文件的随机的一行" lineNum = 0 selected_line = '' while 1: aLine = file_object.readline( ) if not aLine: break lineNum = lineNum + 1 # 本行有多大可能性是文件最后一行? if random.uniform(0,lineNum)<1: #译注:这里请思考1分钟(天才除外 :) selected_line = aLine #译注: 算法的解释见讨论部分 file_object.close( ) return selected_line
讨论 Discussion
当然,更明了的方法是这样的:
random.choice(file_object.readlines( ))
但是,这需要将全部文件内容读入内存,对确实很大的文件可能有问题。
解决部分中算法的理论依据不是很明显,但是也不是太深奥: 当读取到文件第N行时,当前行(即变量selected_line指向的行)被随机取得的可能性是1/N。 容易看出算法选择当前最后读取一行的可能性是对的,因为此处我们以可能性1.0/lineNum的选择最后一行。
由归纳法可以证明算法的合理性:如果读取第N-1行时,选择N-1行的可能性是对的,那么显然对于第N行时算法也是对的。 由于(仅读取到第一行)选择第一行的可能性是1, 因此,算法的随机选择处理直到第N行都是对的。
Of course, the same technique holds for a uniform-probability selection from any finite sequence that, for whatever reason, is made available only one item at a time. But, apart from, for example, selecting a random word from /usr/dict/words, there aren't all that many practical applications of this pretty theorem.
同样的技巧对于从有限序列中,应用均匀分布可能性,一次处理一个元素(原因未知)地随机选择一个元素也是适用的。但是除了从比如/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.