Update marcap_utils.py

FinanceData · Jun 28, 2021 · 2949404 · 2949404
1 parent 3c37b00
commit 2949404
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/marcap_utils.py b/marcap_utils.py
@@ -17,11 +17,11 @@ def marcap_data(start, end=None, code=None):
   end = start if end==None else pd.to_datetime(end)
   df_list = []
 
-  dtypes={'Code':str, 'Name':str, 
-          'Open':int, 'High':int, 'Low':int, 'Close':int, 'Volume':int, 'Amount':float,
-          'Changes':int, 'ChangeCode':str, 'ChagesRatio':float, 'Marcap':float, 'Stocks':int,
-          'MarketId':str, 'Market':str, 'Dept':str,
-          'Rank':int}
+  dtypes={'Code':np.str, 'Name':np.str, 
+          'Open':np.int64, 'High':np.int64, 'Low':np.int64, 'Close':np.int64, 'Volume':np.int64, 'Amount':np.int64,
+          'Changes':np.int64, 'ChangeCode':np.str, 'ChagesRatio':np.float64, 'Marcap':np.int64, 'Stocks':np.int64,
+          'MarketId':np.str, 'Market':np.str, 'Dept':np.str,
+          'Rank':np.int64}
 
   for year in range(start.year, end.year + 1):
     try: