python - Python：linecache.getline 未按预期工作

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 似乎可能是问题所在。linecache.getline(file, lineno.)如果我只是在当前目录中的 1 个 .txt 文件上运行脚本，我就可以成功使用。

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)