lattice/pyerrors
1
0
Fork 0
mirror of https://github.com/fjosw/pyerrors.git synced 2025-03-15 14:50:25 +01:00
Code Issues Projects Releases Packages Wiki Activity

feat: kwargs added to pandas sql methods

Browse source
This commit is contained in:
Fabian Joswig 2022-07-06 11:34:26 +01:00
parent de51d6ca07
commit 18b3672b2d
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
pyerrors/input/pandas.py
View file

@ -7,7 +7,7 @@ from ..correlators import Corr
from .json import create_json_string, import_json_string
def to_sql(df, table_name, db, if_exists='fail', gz=True):
def to_sql(df, table_name, db, if_exists='fail', gz=True, **kwargs):
"""Write DataFrame including Obs or Corr valued columns to sqlite database.
Parameters
@ -25,11 +25,11 @@ def to_sql(df, table_name, db, if_exists='fail', gz=True):
"""
se_df = _serialize_df(df, gz=gz)
con = sqlite3.connect(db)
se_df.to_sql(table_name, con, if_exists=if_exists, index=False)
se_df.to_sql(table_name, con, if_exists=if_exists, index=False, **kwargs)
con.close()
def read_sql(sql, db, auto_gamma=False):
def read_sql(sql, db, auto_gamma=False, **kwargs):
"""Execute SQL query on sqlite database and obtain DataFrame including Obs or Corr valued columns.
Parameters
@ -43,7 +43,7 @@ def read_sql(sql, db, auto_gamma=False):
the error analysis. Default False.
"""
con = sqlite3.connect(db)
extract_df = pd.read_sql(sql, con)
extract_df = pd.read_sql(sql, con, **kwargs)
con.close()
return _deserialize_df(extract_df, auto_gamma=auto_gamma)