在Python中运行R脚本：如何获取所有变量而非仅最后一个？

问题描述 #

我现在用rpy2在Python里运行简单的R脚本，代码如下：

import rpy2.robjects as robjects
r=robjects.r
output = r.source("sample.R")

执行print(output)的时候，只得到脚本最后一个变量的输出，这和我预期的不一样。另外我本来以为直接调用R脚本里定义的c或者data这类变量，Python能直接识别输出，但实际上识别不了，我不知道该怎么调用所有这些变量。

解决方案 #

这其实是R本身的特性导致的——R脚本执行时默认只会返回最后一个表达式的结果，rpy2只是继承了这个行为。下面给你两种可行的解决方法：

方法1：修改R脚本，主动打包需要返回的变量

在你的sample.R脚本末尾，把所有需要导出的变量打包成一个列表（或者数据框），这样r.source()就会返回这个包含所有变量的集合：

# 假设你的脚本里定义了c、data、result三个变量
c <- c(1,2,3)
data <- mtcars[1:5, ]
result <- mean(c)

# 把需要的变量打包成列表返回
list(c_values = c, dataset = data, mean_result = result)

然后在Python里就能直接获取这个列表，按需访问变量：

import rpy2.robjects as robjects
r = robjects.r
output = r.source("sample.R")

# 提取所有变量的集合
all_vars = output[0]
# 查看所有变量名
print("所有变量名：", all_vars.names)
# 访问具体变量
print("c变量的值：", all_vars['c_values'])
print("data变量的值：", all_vars['dataset'])

方法2：直接从R的全局环境读取变量

如果不想修改R脚本，可以通过robjects.globalenv访问R的全局环境，里面包含了脚本执行后所有定义的变量：

import rpy2.robjects as robjects
r = robjects.r
r.source("sample.R")

# 直接读取R全局环境中的变量
c_var = robjects.globalenv['c']
data_var = robjects.globalenv['data']
print("c变量：", c_var)
print("data变量：", data_var)

# 列出全局环境中所有变量名
all_var_names = robjects.globalenv.keys()
print("全局环境所有变量：", all_var_names)

注意事项 #

• 要注意变量名冲突：比如R内置函数c()和你定义的变量c同名时，全局环境里的c会覆盖内置函数，访问时要小心
• 如果脚本变量较多，推荐用方法1，因为主动打包变量更清晰，也避免污染R的全局环境

内容的提问来源于stack exchange，提问作者Jaws