在Python中遍历文件对象不起作用，但是readlines（）可以，但是效率低下

Question

在以下代码中，如果我使用：

for line in fin:

它仅针对“ a”执行

但是，如果我使用：

wordlist = fin.readlines()
for line in wordlist:

然后执行一遍z。

但是readlines()一次读取整个文件，我不希望这样。

如何避免这种情况？

def avoids():
    alphabet = 'abcdefghijklmnopqrstuvwxyz'
    num_words = {}

    fin = open('words.txt')

    for char in alphabet:
      num_words[char] = 0
      for line in fin:
        not_found = True
        word = line.strip()
        if word.lower().find(char.lower()) != -1:
          num_words[char] += 1
    fin.close()
    return num_words

Answer 1

for line in fin的语法只能使用一次。 完成此操作后，您已经用尽了文件，除非您通过fin.seek(0) “重置文件指针”，否则无法再次读取文件。 相反， fin.readlines()将为您提供一个列表，您可以反复遍历。

---

我认为使用Counter （python2.7 +）进行简单的重构可以为您省去麻烦：

from collections import Counter
with open('file') as fin:
    result = Counter()
    for line in fin:
        result += Counter(set(line.strip().lower()))

它将计算文件中包含特定字符的单词数（每行1个单词）（我相信这是您的原始代码...如果我错了，请更正我）

您也可以使用defaultdict （python2.5 +）轻松完成此操作：

from collections import defaultdict
with open('file') as fin:
    result = defaultdict(int)
    for line in fin:
        chars = set(line.strip().lower())
        for c in chars:
            result[c] += 1

最后，把它踢得很老套-我什至不知道什么时候引入了setdefault ...：

fin = open('file')
result = dict()
for line in fin:
    chars = set(line.strip().lower())
    for c in chars:
        result[c] = result.setdefault(c,0) + 1

fin.close()

Answer 2

您有三种选择：

1. 无论如何都读取整个文件。
2. 尝试再次遍历文件之前，请先回到文件的开头。
3. 重新设计代码，以使它不需要多次遍历文件。

Answer 3

尝试：

from collections import defaultdict
from itertools import product

def avoids():
    alphabet = 'abcdefghijklmnopqrstuvwxyz'

    num_words = defaultdict(int)

    with open('words.txt') as fin:
        words = [x.strip() for x in fin.readlines() if x.strip()]

    for ch, word in product(alphabet, words):
        if ch not in word:
             continue
        num_words[ch] += 1

    return num_words