我是 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
答案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
如果不合适,很乐意进行编辑。