Browse Source

list_judge = list(set(data.keys()) & set(stock_list))减少计算量 提升效率

Daniel 2 years ago
parent
commit
85c2c7a752
1 changed files with 4 additions and 2 deletions
  1. 4 2
      QMT/real_time.py

+ 4 - 2
QMT/real_time.py

@@ -70,8 +70,10 @@ def his_vol(stock, num):
 
 
 def ma_judge(data, stock_list, results):
-    print('这个ma_judge的PID为:', os.getpid())
-    for stock in data:
+    print(f'这个ma_judge的PID为:{os.getpid()},收到的data数据为:{len(data.keys())}')
+    list_judge = list(set(data.keys()) & set(stock_list))
+    print(len(list_judge), list_judge)
+    for stock in list_judge:
         i = stock.replace('XSHG', 'SH').replace('XSHE', 'SZ')
         current_price, open_price = data[i]['lastPrice'], data[i]['open']
         MA5, MA10, MA20 = ma(i, 5, data), ma(i, 10, data), ma(i, 20, data)