Pandas Dataframe to Code

Question

既存のpandasデータフレームがある場合は、別のPythonスクリプトで実行するとそのデータフレームを再現するpythonコードを生成する方法がありますか？

In[1]: df 

Out[1]: 
    income user 
0 40000 Bob 
1 50000 Jane 
2 42000 Alice 

In[2]: someFunToWriteDfCode(df) 

Out[2]: 
df = pd.DataFrame({'user': ['Bob', 'Jane', 'Alice'], 
    ...:     'income': [40000, 50000, 42000]}) 

Answer 1

最初にデータフレームを保存し、必要に応じて別のPythonスクリプトを読み込むことができます。 2つのパッケージ（pickleとshelve）で実行できます。

import pandas as pd 
import pickle 
df = pd.DataFrame({'user': ['Bob', 'Jane', 'Alice'], 
        'income': [40000, 50000, 42000]}) 
with open('dataframe', 'wb') as pfile: 
    pickle.dump(df, pfile)   # save df in a file named "dataframe" 

を別のファイルにデータフレームを読み取るには：pickleでそれを行うには

import pickle 
with open('dataframe', 'rb') as pfile: 
    df2 = pickle.load(pfile)  # read the dataframe stored in file "dataframe" 
    print(df2) 

が出力：

income user 
0 40000 Bob 
1 50000 Jane 
2 42000 Alice 

shelveでそれを行うには：

別のファイルにデータフレームを読み取るために

import pandas as pd 
import shelve 
df = pd.DataFrame({'user': ['Bob', 'Jane', 'Alice'], 
        'income': [40000, 50000, 42000]}) 
with shelve.open('dataframe2') as shelf: 
    shelf['df'] = df    # store the dataframe in file "dataframe" 

：

import shelve 
with shelve.open('dataframe2') as shelf: 
    print(shelf['df'])    # read the dataframe 

が出力：場合

print "df = pd.DataFrame(%s)" % (str(df.to_dict())) 

：

income user 
0 40000 Bob 
1 50000 Jane 
2 42000 Alice 

Answer 2

あなたはDATAFRAME上to_dict（）メソッドを使用することを試みることができますデータにNaNが含まれている場合は、float（ 'nan'）で置き換える必要があります：

print "df = pd.DataFrame(%s)" % (str(df.to_dict()).replace(" nan"," float('nan')"))