代码在函数外部工作,但不在函数内部(python)

时间:2017-11-13 03:06:27

标签: python function for-loop import

我尝试打印文件中的行数,当我使用函数外的代码时,它运行正常:

input_file_name = input("Please Enter the name of your text file: ")
infile = open(input_file_name, "r")

listOfLines = infile.readlines()
count = 0
char = " "

for line in listOfLines :
    text = line.rstrip()
    while char != "":
        char = infile.read(1)
    count = count + 1
print(count)


infile.close()

但是当我在函数内部使用相同的代码时,它会输出值0而不是5(这是我的测试文件的长度)。

def main():

    input_file_name = input("Please Enter the name of your text file: ")
    infile = open(input_file_name, "r")
    print()

    print_file(infile)
    count_lines(infile)

    infile.close()


def print_file(infile):
    listOfLines = infile.readlines()
    for line in listOfLines:
        text = line.rstrip()
        print(text)

def count_lines(infile):
    listOfLines = infile.readlines()
    count = 0
    char = " "

    for line in listOfLines :
        text = line.rstrip()
        while char != "":
            char = infile.read(1)
        count = count + 1
    print(count)

main()

但是,如果我删除了该功能

def print_file(infile):
listOfLines = infile.readlines()
for line in listOfLines:
    text = line.rstrip()
    print(text)

以及对函数的调用,

print_file(infile)

它工作正常。为什么第一个函数会影响第二个函数的打印方式?

1 个答案:

答案 0 :(得分:2)

使用infile = open(input_file_name, "r")创建的文件对象是迭代器。它只能循环一次。

您的print_file函数对文件执行第一次迭代,耗尽其内容。当你在count_lines中尝试第二次迭代时,迭代器已经结束,所以没有其他事情发生。