diff options
Diffstat (limited to 'sql_functions.py')
| -rw-r--r-- | sql_functions.py | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/sql_functions.py b/sql_functions.py new file mode 100644 index 0000000..313c449 --- /dev/null +++ b/sql_functions.py @@ -0,0 +1,18 @@ +import sqlite3 as sql +import pandas as pd + +def execute_sql_statement(sql_statement): + conn = sql.connect("database.db") + cur = conn.cursor() + cur.execute(sql_statement) + rows = cur.fetchall() + return rows + +def run_sql_pandas(sql_statement): + conn = sql.connect("database.db") + df=pd.read_sql_query(sql_statement, conn).to_records(index=False) + return df + +def get_list_of_dict(keys, list_of_tuples): + list_of_dict = [dict(zip(keys, values)) for values in list_of_tuples] + return list_of_dict
\ No newline at end of file |
