dsyoon/DeepStock
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

init

Browse Source
This commit is contained in:
dsyoon
2021-10-19 11:19:39 +09:00
parent 0462c0c46a
commit 8dda63f779
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
hts/OrderChecker.py
View File

@@ -18,9 +18,10 @@ class OrderChecker:
else: else:
order_df = pd.DataFrame(columns=["stock_code", "type", "orderNum", "count", "price", "datetime"]) 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: if 'Unnamed: 0' in order_df.columns:
order_df.drop(['Unnamed: 0'], axis=1, inplace=True) 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']) order_df['datetime'] = pd.to_datetime(order_df['datetime'])
return order_df return order_df