添加指数名称

chart.py

@@ -182,12 +182,21 @@ def resample_data(df: pd.DataFrame, timeframe: str) -> pd.DataFrame:
 if __name__ == "__main__":
     code = "399986"
     timeframe = "1W"
+
+    df = pd.read_csv(
+        "/Users/aszer/Documents/vscode/etf/data/index_all_stock.csv",
+        encoding="utf-8-sig",
+    )
+    name = df.loc[df["代码"] == code, "名称"].values[0]
+
     df = get_kline(code=code)
     df = resample_data(df, timeframe)
     chart = Chart(toolbox=True, inner_height=0.7, maximize=True)
 
     chart.topbar.textbox("symbol", code)
+    chart.topbar.textbox("name", name)
     chart.topbar.textbox("timeframe", timeframe)
+
     # chart.time_scale(visible=False) # 将主图的时间轴隐藏
     chart.legend(visible=True)