优化财报获取

财报筛选/loader.py

@@ -0,0 +1,22 @@
+import pandas as pd
+
+finance_root = "C:\\Users\\lanyuanxiaoyao\\Documents\\Tushare"
+
+
+def load_balance_sheet():
+    return pd.read_csv(f"{finance_root}\\财务报表\\资产负债表1990-2024.csv")
+
+
+def load_income():
+    return pd.read_csv(f"{finance_root}\\财务报表\\利润表1990-2024.csv")
+
+
+def load_cashflow():
+    return pd.read_csv(f"{finance_root}\\财务报表\\现金流量表1990-2024.csv")
+
+
+def load_finance():
+    finance_df = load_balance_sheet()
+    finance_df = pd.merge(finance_df, load_income(), on=["ts_code", "end_date"])
+    finance_df = pd.merge(finance_df, load_cashflow(), on=["ts_code", "end_date"])
+    return finance_df