计算多个文件中的行数并连同文件名一起输出

Question

我想获取文件夹中每个文件的行数，然后将行数与文件名相邻打印出来。刚刚进入编程世界，我设法编写了这段简短的代码，到处借用它们。

#count the number of lines in all files and output both count number and file name
import glob
list_of_files = glob.glob('./*.linear')
for file_name in list_of_files:
    with open (file_name) as f, open ('countfile' , 'w') as out :
        count = sum (1 for line in f)
        print >> out, count, f.name

但这只会输出其中一个文件。

这可以很容易地在 shell 中使用 wc -l *.linear 完成，但我想知道如何在 python 中做到这一点。

P.S : 我真诚地希望我没有重复问题！

Answer 1

你真的很亲密！只需打开一次计数文件，而不是在循环内：

import glob
with open('countfile' , 'w') as out:
    list_of_files = glob.glob('./*.linear')
    for file_name in list_of_files:
        with open(file_name, 'r') as f:
            count = sum(1 for line in f)
            out.write('{c} {f}\n'.format(c = count, f = file_name))

每次以w 模式打开文件时（例如open('countfile', 'w')），countfile 的内容（如果已存在）都会被删除。这就是为什么你只需要调用一次。