[Mod] 优化循环逻辑

vnpy · Sep 29, 2021 · 106715e · 106715e
1 parent 8195ed6
commit 106715e
Showing 1 changed file with 6 additions and 8 deletions.
diff --git a/vnpy_dolphindb/dolphindb_database.py b/vnpy_dolphindb/dolphindb_database.py
@@ -230,17 +230,16 @@ def load_bar_data(
         )
 
         df.set_index("datetime", inplace=True)
-        ds = df.tz_localize(DB_TZ)
+        df = df.tz_localize(DB_TZ)
 
         # 转换为BarData格式
         bars: List[BarData] = []
 
-        for tp, dt in zip(ds.itertuples(), ds.index):
-
+        for tp in df.itertuples():
             bar = BarData(
                 symbol=symbol,
                 exchange=exchange,
-                datetime=dt.to_pydatetime(),
+                datetime=tp.Index.to_pydatetime(),
                 interval=interval,
                 volume=tp.volume,
                 turnover=tp.turnover,
@@ -283,17 +282,16 @@ def load_tick_data(
         )
 
         df.set_index("datetime", inplace=True)
-        ds = df.tz_localize(DB_TZ)
+        df = df.tz_localize(DB_TZ)
 
         # 转换为TickData格式
         ticks: List[TickData] = []
 
-        for tp, dt in zip(ds.itertuples(), ds.index):
-
+        for tp in df.itertuples():
             tick = TickData(
                 symbol=symbol,
                 exchange=exchange,
-                datetime=dt.to_pydatetime(),
+                datetime=tp.Index.to_pydatetime(),
                 name=tp.name,
                 volume=tp.volume,
                 turnover=tp.turnover,