From 8dda63f779422d1f4dbd3c9d702e9f1210c1e426 Mon Sep 17 00:00:00 2001 From: dsyoon Date: Tue, 19 Oct 2021 11:19:39 +0900 Subject: [PATCH] init --- hts/OrderChecker.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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