python - 在python项目中使用相对路径读取文件

Question

假设我有一个结构如下的 python 项目:

project
    /data
        test.csv
    /package
        __init__.py
        module.py
    main.py

__init__.py :

from .module import test

module.py :

import csv

with open("..data/test.csv") as f:
    test = [line for line in csv.reader(f)]

main.py :

import package

print(package.test)

当我运行 main.py我收到以下错误:

 C:\Users\Patrick\Desktop\project>python main.py
Traceback (most recent call last):
  File "main.py", line 1, in <module>
    import package
  File "C:\Users\Patrick\Desktop\project\package\__init__.py", line 1, in <module>
    from .module import test
  File "C:\Users\Patrick\Desktop\project\package\module.py", line 3, in <module>
    with open("../data/test.csv") as f:
FileNotFoundError: [Errno 2] No such file or directory: '../data/test.csv'

但是，如果我运行 module.py来自 package目录我没有错误。所以看来open(...)中使用的相对路径仅相对于原始文件的运行位置(即 __name__ == "__main__" )？我不想使用绝对路径。有什么方法可以解决这个问题？

Answer 1

相对路径是相对于 current working directory .
如果你不希望你的路径是，它必须是绝对的。
但是有一个常用的技巧来从当前脚本构建绝对路径:使用它的 __file__ 特殊属性:

from pathlib import Path

path = Path(__file__).parent / "../data/test.csv"
with path.open() as f:
    test = list(csv.reader(f))

这需要 python 3.4+(对于 pathlib 模块)。
如果您仍然需要支持旧版本，您可以通过以下方式获得相同的结果:

import csv
import os.path

my_path = os.path.abspath(os.path.dirname(__file__))
path = os.path.join(my_path, "../data/test.csv")
with open(path) as f:
    test = list(csv.reader(f))

[ 2020年编辑: python3.4+ 现在应该是规范，所以我首先移动了受 jpyams 评论启发的 pathlib 版本]