Question

在目录的底部有一些.txt文件，我需要从中提取第2行。

import os
import os.path
import linecache

for dirpath, dirnames, filenames in os.walk("."):
    for filename in [f for f in filenames if f.endswith(".txt")]:
        #print os.path.join(dirpath, filename)
        #print filename
        print linecache.getline(filename, 2)

我能够成功解析所有目录并找到每个文本文件。但是linecache.getline只返回换行符（应该有来自该行文件的数据）。使用

print linecache.getline(filename, 2).rstrip('\n')

也解决不了这个问题。

我能够正确地打印出每个目录中的文件名，但是将这些文件名传递给linecache似乎可能是个问题。如果我只是在当前目录中的1 .txt文件上运行脚本，我就能成功使用linecache.getline(file, lineno.)。

Answer 1

linecache.getline从当前工作目录中获取文件名。

因此解决方案：

import os
import os.path
import linecache

for dirpath, dirnames, filenames in os.walk("."):
    for filename in [f for f in filenames if f.endswith(".txt")]:
        direc = os.path.join(dirpath, filename)
        print linecache.getline(direc, 2)

Python：linecache.getline没有按预期工作

1 个答案: