如何避免在Python脚本中使用路径名？

Question

在我的Python脚本中，我编写了特定的路径名​​。如果我在另一台计算机或文件夹上使用它，它将无法工作。

ex：df = pd.read_csv('C:/VerySpecificNameForDocuments/test.csv', 'wb')

我见过使用：%/...

如何重新编写此代码才能在其他计算机/目录中使用？

Answer 1

如果您不想在脚本上hardcode path，environment variables是您的朋友，在python中您可以使用os.environ访问它们，即：

import os
# %USERPROFILE% environment variable  + "/documents" creates the path to `My documents` folder on windows systems
csv_file =  '{}\\documents\\file.csv'.format(os.environ['USERPROFILE'])
# C:\Users\Administrador\documents\file.csv
df = pd.read_csv(csv_file, 'wb')

其他有趣的窗口environment variables：

>>> import os
>>> os.environ['HOMEPATH']
'\\Users\\Administrador'
>>> os.environ['HOMEDRIVE']
'C:'
>>> os.environ['WINDIR']
'C:\\WINDOWS'
>>> os.environ['APPDATA']
'C:\\Users\\Administrador\\AppData\\Roaming'
>>>

PS：
如果您需要支持多个操作系统（linux，mac，windows），您可能需要实现一个简单的OS detection功能并单独处理每个系统。