通过rpy2传递列表作为参数

2024-09-24 00:29:13 发布

您现在位置:Python中文网/ 问答频道 /正文

我在Python中使用R的spi函数。其中一个参数(“kernel”)需要一个列表,我不知道在使用rpy2时如何传递它。R语法如下(带有的“kernel”参数):

library("lmomco", lib.loc="~/lib/R_libs")
library("SPEI", lib.loc="~/lib/R_libs")

data = read.csv("rainfall_data.csv", header = FALSE)
result = spi(data, scale=10, distribution='Gamma', fit='max-lik', kernel = list(type='rectangular', shift=0))

到目前为止,我的Python脚本(没有使用“kernel”参数)如下所示:

import numpy as np 
from rpy2.robjects.packages import importr
importr('lmomco')
SPEI = importr('SPEI')

import rpy2.robjects.numpy2ri as numpy2ri
numpy2ri.activate()

data = np.genfromtxt('rainfall_data.csv', delimiter=',')

result = np.array(SPEI.spi(data, scale=10, distribution='Gamma', fit='max-lik')[1])

如何将内核作为rpy2的列表传递?你知道吗


Tags: csvimportspi列表data参数libnp
1条回答
网友
1楼 · 发布于 2024-09-24 00:29:13

考虑从python字典创建ListVector,然后将其传递到方法中:

import rpy2.robjects as robjects 
...
x = robjects.ListVector({'type': 'rectangular', 'shift': 0})

result = np.array(SPEI.spi(data, scale=10, distribution='Gamma', fit='max-lik', kernel=x)[1])

相关问题 更多 >