api: 修改对 numpy 数据类型的判断，以覆盖更多类型

如 numpy.int 是 numpy.integer 下的子类型，不能覆盖到与它在同一个类型层次中的类型.因此若要判断是否是整型，应使用 integer close #180
shinnytech · Dec 19, 2019 · 279a3ca · 279a3ca
1 parent 1f4ab5a
commit 279a3ca
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tqsdk/api.py b/tqsdk/api.py
@@ -1230,13 +1230,13 @@ def _update_serial_single(self, serial):
                 array[0:serial["width"] - shift] = array[shift:serial["width"]]
                 for ext in serial["extra_array"].values():
                     ext[0:serial["width"] - shift] = ext[shift:serial["width"]]
-                    if ext.dtype == np.float:
+                    if np.issubdtype(ext.dtype, np.floating):
                         ext[serial["width"] - shift:] = np.nan
-                    elif ext.dtype == np.object:
+                    elif np.issubdtype(ext.dtype, np.object_):
                         ext[serial["width"] - shift:] = None
-                    elif ext.dtype == np.int:
+                    elif np.issubdtype(ext.dtype, np.integer):
                         ext[serial["width"] - shift:] = 0
-                    elif ext.dtype == np.bool:
+                    elif np.issubdtype(ext.dtype, np.bool_):
                         ext[serial["width"] - shift:] = False
                     else:
                         ext[serial["width"] - shift:] = np.nan