在哪里将file.close（）放入python循环中

Question

我正在尝试制作一个简单的程序，该程序应该在文本文件中搜索单词“为什么”并打印出现的次数。 这是代码：

def get():
    a = 0
    target = str(raw_input('name file > '))
    file = open(target,'r+')
    main(file)

def main(target):
    for x in target:
        if x == 'why':
           a+= 1 

    print a

file.close()

get()

但是我应该把file.close()放在file.close() ？ 我需要将它放在main（）的for循环中还是可以将其放在代码的末尾？

Answer 1

使用完文件后，执行file.close() 。 所以在这里，最好是在for循环之后执行。

def main(target):for x in target:
    if x == 'why':
        a+= 1 
    file.close()
    print a

另外，您也可以将with open('file.txt', 'r+') as f来自动关闭文件（并且更具Python风格）：

def get():
    a = 0
    target = str(raw_input('name file > '))
    with open(target,'r+') as myfile:
        main(myfile)

def main(target):
    for x in target:
        if x == 'why':
            a+= 1  
    print a

get()

所以在这里，在main（）完成之后，文件将自动关闭。

---

请注意，您不应该为变量file命名，因为该file已经是内置函数，因此您将对其进行覆盖。

同样，也无需在raw_input() :)周围调用str() 。 raw_input已经返回一个字符串。

---

最后，您的错误实际上会引发UnboundLocalError，因为a仅在get()函数中定义。 如果要在其他函数中访问它，请在定义变量:)之前放入global a 。

Answer 2

管理打开file对象的Python方式是使用with语句 。 这将在块末尾关闭file 。 由于file是Python内置的（并且您不想追加到文件中），因此get()函数的固定版本可能如下所示：

def get():
    filename = raw_input('name file > ')
    with open(filename, 'r') as source:
        main(source)

需要注意的是x == 'why'将始终评估为False ，因为for x in target在每行迭代target ，留在换行符。 如果要比较整行，则需要将其strip 。 您的main()固定版本可能如下所示：

def main(target):
    print sum(line.strip() == 'why' for line in target)

如果要count每个事件why在每一行，然后sum(line.count('why') for line in target)可能是一个更好的办法。

最后，如果要import此文件（例如，进入一组单元测试！），则应使用if __name__ == '__main__':保护对（名称不明确的） get()函数的if __name__ == '__main__':

if __name__ == '__main__':
    get()