从文本文件中完成打印N行

Question

我是 Python 新手，我想编写一个脚本，打印出我作为用户输入的目录中每个文本文件的前 N ​​行。我无法完成这项工作。任何帮助都将不胜感激。

我输入的文件路径是 /home/myprofile/

#!/usr/bin/env python

import glob, os
dirpath=input("Enter path: ")
for file in glob.glob(dirpath + "*.txt" ):
        print(file)
        with open("file") as myfile:
        head = [next(myfile) for 2 in xrange(2)]
        print head

Answer 1

这对你有用吗：

#!/usr/bin/env python

import glob, os
dirpath=raw_input("Enter path: ")

for file in glob.glob(dirpath+"/*.txt" ):
    print(file)
    with open(file) as myfile:
        for x in range(0, 5):
            print myfile.readline().rstrip() #rstrip to remove \n

如果不合适，很乐意进行编辑。