使用rpy2无法将R分析输出结果转回为Python DataFrame问题求助

1 查看R全局环境的操作 #

你可以通过以下代码查看R全局环境中的对象及结构，定位返回值类型问题：

# 查看R全局环境中所有变量名
print(base.ls())
# 查看results对象的完整结构（对应R中的str()函数）
print(base.str(results))
# 查看results返回列表的键名
print(results.names)
# 分别查看两个返回值的类型
print("params对象类型：", base.class(results[0]))
print("r2对象类型：", base.class(results[1]))

2 方案一：直接转换为pandas DataFrame #

你之前转换失败的核心原因是：summary(fit)返回的是nls拟合结果的S3类对象，不是R原生数据框，无法直接强制转换为pandas对象。你可以先提取其中的系数矩阵再转换，示例代码如下：

import pandas as pd
from rpy2.robjects import pandas2ri
pandas2ri.activate()

# 提取拟合系数表，转换为pandas DataFrame
coef_df = robjects.conversion.rpy2py(results.rx2("params").rx2("coefficients"))
# 提取r2值
r2_val = results.rx2("r2")[0]
# 将r2加入系数表中，也可以单独存储
coef_df["r_squared"] = r2_val

# 查看转换后的结果
print(coef_df)
print(type(coef_df))

3 方案二：导出为csv后导入Python #

你之前导出失败是因为直接把嵌套列表对象传入write.csv，需要先将系数转换为数据框再导出：

# R侧将系数表转成数据框后导出csv
robjects.r('''
fit_coef <- as.data.frame(summary(results)$coefficients)
fit_coef$r_squared <- results$r2
write.csv(fit_coef, "curve_fit_results.csv", row.names = TRUE)
''')

# Python侧读取csv
df = pd.read_csv("curve_fit_results.csv", index_col=0)
print(df)

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