In [11]:
# pickle test
import pickle
import pandas as pd
import os
import zlib
this_dir = os.getcwd()
mydata = {'col1':[1,2,3],'col2':[4,5, 6], 'col3':['abc','def','ghi']}
one_list = [1,2,3,4]
X = 5.678
In [12]:
df = pd.DataFrame(mydata)
df
Out[12]:
In [13]:
pickle_jar = {}
pickle_jar['mydata'] = df
pickle_jar['alist'] = one_list
pickle_jar['variable'] = X
pickle_jar
Out[13]:
In [14]:
with open(this_dir + '/mypickle.py', 'wb') as pickle_file:
#pkle_string = zlib.compress(pickle.dumps(pickle_jar))
#pickle_file.write(pkle_string)
pickle_file.write(zlib.compress(pickle.dumps(pickle_jar)))
pickle_file.close()
In [14]: