Python調(diào)用R代碼

import pandas as pd
import rpy2.robjects as robjects
#pandas和r中的dataframe類(lèi)型轉(zhuǎn)換需要用到pandas2ri  
from rpy2.robjects import pandas2ri   

pandas2ri.activate()
robjects.r(''''' 
     library(dplyr)  # 導(dǎo)入包測(cè)試
     ''')  

caculate = robjects.r(''''' 
       f <- function(dat, fun){
              x = dat[,1]
              y = dat[,2]
              if(fun=='add'){
                result = x+y
              }
              if (fun=='division'){
                result = x/y
              }
              return(result)
       } 
       ''')   # 自定義函數(shù)
#通過(guò)pandas2ri.py2ri將pandas的dataframe轉(zhuǎn)成r作為輸入，再使用pandas2ri.ri2py將輸出的r中的dataframe轉(zhuǎn)成pandas          
dat = pd.DataFrame({'x':[1,2,3], 'y':[2,2,2]})
result = pandas2ri.ri2py((caculate(pandas2ri.py2ri(dat), "add")))  # [3, 4, 5]
result = pandas2ri.ri2py((caculate(pandas2ri.py2ri(dat), "division")))  # [0.5, 1. , 1.5]
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成，瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明：文章內(nèi)容（如有圖片或視頻亦包括在內(nèi)）由作者上傳并發(fā)布，文章內(nèi)容僅代表作者本人觀點(diǎn)，簡(jiǎn)書(shū)系信息發(fā)布平臺(tái)，僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容