#158. Add missing function

open-genes · Sep 10, 2023 · 009c706 · 009c706
1 parent a4cd250
commit 009c706
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/scripts/expression_change_human_mrna/replace.py b/scripts/expression_change_human_mrna/replace.py
@@ -82,6 +82,14 @@
     ),
 )
 
+def get_df_from_db(connection: MySQLConnection, table: str, columns: List) -> DataFrame:
+    """Read data from Database from provided table with provided columns"""
+    columns = ", ".join(columns)
+    sql = f"SELECT {columns} FROM {table}"
+    df = pd.read_sql(sql, con=connection)
+    df = df.dropna()
+    return df
+
 def get_df_from_csv(file_name: str) -> DataFrame:
     """Read csv file by provided file name"""
     cur_dir = os.path.dirname(os.path.abspath(__file__))