lattice/pyerrors
1
0
Fork 0
mirror of https://github.com/fjosw/pyerrors.git synced 2025-05-14 19:43:41 +02:00
Code Issues Projects Releases Packages Wiki Activity

feat: pandas DataFrames with Corr columns can now also be imported and

Browse source 
exported.
This commit is contained in:
Fabian Joswig 2022-06-30 15:26:31 +01:00
parent c7c1725667
commit 153cc795b8
2 changed files with 14 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
pyerrors/input/pandas.py
View file

@ -2,6 +2,7 @@ import warnings
import gzip
import pandas as pd
from ..obs import Obs
from ..correlators import Corr
from .json import create_json_string, import_json_string
@ -23,7 +24,7 @@ def dump_df(df, fname, gz=True):
out = df.copy()
for column in out:
if isinstance(out[column][0], Obs):
if isinstance(out[column][0], (Obs, Corr)):
out[column] = out[column].transform(lambda x: create_json_string(x, indent=0))
if not fname.endswith('.csv'):
@ -67,6 +68,6 @@ def load_df(fname, auto_gamma=False, gz=True):
if re_import[column][0][:20] == '{"program":"pyerrors':
re_import[column] = re_import[column].transform(lambda x: import_json_string(x, verbose=False))
if auto_gamma is True:
re_import[column].apply(Obs.gamma_method)
re_import[column].apply(lambda x: x.gamma_method())
return re_import