python:文件路径不再适用于 imp

Question

我最近开始使用导入来更好地组织我在 python 中的代码。我在 file1.py 中的原始代码使用了以下行:

def foo():
    files = [f for f in os.listdir('.') if os.path.isfile(f)]
    print files
    #do stuff here....

它引用了与代码相同的文件夹中的所有文件，print files 将正确的输出显示为文件名数组。

但是，我最近将目录结构更改为如下所示:

./main.py
./folder1/file1.py
./folder1/data_file1.csv
./folder1/data_file2.csv
./folder1/......

在 main.py 中，我使用:

import imp
file1 = imp.load_source('file1', "./folder1/file1.py")
.
.
.

file1.foo()

现在，files 是一个空数组。发生了什么？除了相对路径之外，我还尝试过绝对文件路径。直接使用 data_file1.csv 声明一个数组是可行的，但我无法获得任何其他方法来处理此导入。

这是怎么回事？

Answer 1

当你执行 os.listdir('.') 时，你试图列出 '.' (这是当前目录)的内容，但是这不需要是脚本所在的目录，它将是您运行脚本时所在的目录(除非您在 python 脚本中使用了 os.chdir())。

你不应该依赖于当前的工作目录，而是可以使用__file__ 来访问当前脚本的路径，然后你可以使用os.path.dirname()获取脚本所在的目录。

然后你可以使用os.path.join()将从 os.listdir() 获得的文件路径与脚本目录连接起来，并检查这些路径是否是创建文件列表的文件。

例子-

def foo():
    filedir = os.path.dirname(__file__)
    files = [f for f in (os.path.join(filedir, fil) for fil in os.listdir(filedir)) if os.path.isfile(f)]
    print files
    #do stuff here....