函数仅在第一次调用时才能正常工作[重复]

Question

我有以下功能，第二个功能count_forbid(a)只能工作1次。在此示例中，它计算不包含字母 'c' 的单词的正确值，但对于 y，它返回零。所以这意味着代码只能在第一次正确执行，而在所有其他时间返回零：

import string
fin = open('words.txt')
def forbid_or_not(word,forb):
    for letter in word:
        if letter in forb:
            return False
    return True

def count_forbid(a):
    count = 0
    for line in fin:
        word1 = line.strip()
        if forbid_or_not(word1,a):
            count += 1
    return count

x = count_forbid('c')
y = count_forbid('d')

Answer 1

在你遍历文件之后：

    for line in fin:

它会走到尽头，尝试重新迭代将没有效果。

要么更改函数以使用context manager，在调用函数时重新打开文件：

def count_forbid(a):
    count = 0
    with open('words.txt') as fin:  # closes the file automatically
        for line in fin:
            word1 = line.strip()
            if forbid_or_not(word1,a):
                count += 1
    return count

这是在 python 中打开文件的首选方式。

或者，在您的调用之间添加 fin.seek(0) 以使文件指向开头：

x = count_forbid('c')
fin.seek(0)
y = count_forbid('d')