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