Конвертировать Pandas df в файл rda - PullRequest
Новая
       91

Конвертировать Pandas df в файл rda

1 голос
/ 28 февраля 2020

Я очищаю свои данные в Python, но программа, которую мы используем для визуализаций, настроена для R. Я пытаюсь сохранить свои данные в виде файлов rda. Я смог найти ресурсы здесь, чтобы начать, однако мой df имеет 92 столбца, когда он конвертируется в rda, он имеет 1942 столбца, это похоже на следующее. 

import rpy2
from rpy2 import robjects
from rpy2.robjects import pandas2ri
pandas2ri.activate()

df = pd.DataFrame({'A': [1, 2, 3], 'B':[10, 9, 8], 'C':['A', 'B', 'C']})


    A   B   C
0   1   10  A
1   2   9   B
2   3   8   C

РЕДАКТИРОВАТЬ: я пытался оба этих преобразования и получили одинаковый результат с обоими. 

df = pandas2ri.py2ri(df)

И 

def save_rdata_file(df, filename):
    r_data = pandas2ri.py2ri(df)
    robjects.r.assign('my_df', r_data)
    robjects.r("save(my_df, file=paste0('{}'))".format(filename))

save_rdata_file(df, file_location)



result = pd.DataFrame({'A': [1, 2, 3], 'B':[10, 9, 8], 'C':['A', 'B', 'C'],
                   'A.1': [1, 1, 1], 'A.2': [2, 2, 2], 'A.3':[3, 3, 3],
                   'B.1':[10, 10, 10], 'B.2':[9, 9, 9], 'B.3':[8, 8, 8],
                   'C.1':['A', 'A', 'A'], 'C.2':['B', 'B', 'B'], 'C.3':['C', 'C', 'C']})

    A   B   C   A.1 A.2 A.3 B.1 B.2 B.3 C.1 C.2 C.3
0   1   10  A   1   2   3   10  9   8   A   B   C
1   2   9   B   1   2   3   10  9   8   A   B   C
2   3   8   C   1   2   3   10  9   8   A   B   C

1 Ответ

0 голосов
/ 28 февраля 2020

Я думаю, что вы использовали @ решение anthonybell : 

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

df = pd.DataFrame({'A': [1, 2, 3], 'B':[10, 9, 8], 'C':['A', 'B', 'C']})
r_data = pandas2ri.py2ri(df)
robjects.r.assign("df", r_data)
robjects.r("save(df, file='test.rda')")

In R:
load("test.rda")
> ls()
[1] "df"
> str(df)
'data.frame':   3 obs. of  3 variables:
 $ A: int  1 2 3
 $ B: int  10 9 8
 $ C: chr  "A" "B" "C"

Это работает вполне нормально для меня. Я на R 3.6.1, python 3.6.10, pandas 1.0.1, rpy 2.9.4. Ниже приведено другое решение, использующее рассол, но оно требует использования сетчатки в R: 

import pandas as pd
df = pd.DataFrame({'A': [1, 2, 3], 'B':[10, 9, 8], 'C':['A', 'B', 'C']})
df.to_pickle("./test.pkl")

In R:
library(reticulate)
pd = import("pandas",convert=TRUE)
str(df)
'data.frame':   3 obs. of  3 variables:
 $ A: num  1 2 3
 $ B: num  10 9 8
 $ C: chr  "A" "B" "C"
 - attr(*, "pandas.index")=RangeIndex(start=0, stop=3, step=1)
...