Current section
Files
Jump to
Current section
Files
priv/jun_dataframe.py
import pandas as pd
import numpy as np
from erlport.erlterms import Atom
from erlport.erlang import set_encoder
def setup_dtype():
set_encoder(dtype_encoder)
return Atom("ok")
def dtype_encoder(value):
if isinstance(value, np.int64):
return np.asscalar(value)
elif isinstance(value, np.float64):
return np.asscalar(value)
elif isinstance(value, pd.core.frame.DataFrame):
return (Atom("pandas.core.frame.DataFrame"), value)
else:
return value