diff --git a/hts/OrderChecker.py b/hts/OrderChecker.py index 70ea944..5810d69 100644 --- a/hts/OrderChecker.py +++ b/hts/OrderChecker.py @@ -18,9 +18,10 @@ class OrderChecker: else: order_df = pd.DataFrame(columns=["stock_code", "type", "orderNum", "count", "price", "datetime"]) - order_df = order_df.astype({"stock_code": str, "type": str, "orderNum": int, "count": int, "price": int}) if 'Unnamed: 0' in order_df.columns: order_df.drop(['Unnamed: 0'], axis=1, inplace=True) + order_df = order_df.fillna(0) + order_df = order_df.astype({"stock_code": str, "type": str, "orderNum": int, "count": int, "price": int}) order_df['datetime'] = pd.to_datetime(order_df['datetime']) return order_df