R背景转Python：RStudio中获取脚本目录遇__file__未定义错误

问题场景 #

具备R语言背景，正在学习Python，使用RStudio运行Python代码，想要获取当前Python脚本所在目录，编写代码如下：

import os
result = os.path.join(os.path.dirname(__file__))
print(result)

执行时出现错误：

>>> import os
>>> result = os.path.join(os.path.dirname(__file__))
NameError: name '__file__' is not defined

原因分析 #

__file__是Python内置变量，仅当脚本文件被完整执行（比如通过python script.py命令，或RStudio中"Run Script"执行整个脚本）时才会被定义。如果是在RStudio的Python交互式控制台逐行执行代码，这个变量不存在，因此触发NameError。

另外，代码里的os.path.join仅传单个参数是多余的，os.path.dirname(__file__)本身已经是目录路径，可直接使用。

解决方案 #

方案1：执行完整脚本文件 #

把代码保存为.py格式的脚本文件，在RStudio中通过以下方式执行：

• 点击脚本编辑器右上角的Run Script按钮
• 或在R控制台中用reticulate::source_python("你的脚本路径.py")执行

此时__file__会被正确定义，建议优化代码为：

import os
# 获取脚本所在目录的绝对路径
script_dir = os.path.abspath(os.path.dirname(__file__))
print(script_dir)

方案2：交互式环境下获取路径 #

如果是在RStudio的Python控制台逐行调试，可通过以下方式获取路径：

1. 获取RStudio当前工作目录

   import os
   print(os.getcwd())
   

   注意：这个路径是RStudio的当前工作目录，不一定等于脚本所在目录，可通过RStudio右下角的"Files"面板切换工作目录到脚本所在位置。

2. 获取当前打开的脚本文件路径
   借助R的rstudioapi包获取当前打开的脚本路径，再传给Python：

   • 先在R控制台执行：

     library(rstudioapi)
     script_path <- getActiveDocumentContext()$path
     

   • 再在Python控制台执行：

     import os
     import rpy2.robjects as ro
     # 读取R中获取的脚本路径
     script_path = ro.r('script_path')[0]
     script_dir = os.path.dirname(script_path)
     print(script_dir)
     

内容的提问来源于stack exchange，提问作者bash-asker