fixed DRAWIMG 添加 nan 保护

fasiondog · Feb 12, 2025 · 32f109e · 32f109e
1 parent 243c616
commit 32f109e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/hikyuu/draw/drawplot/matplotlib_draw.py b/hikyuu/draw/drawplot/matplotlib_draw.py
@@ -1288,7 +1288,8 @@ def DRAWIMG(cond: Indicator, price: Indicator, img: str, kdata: KData = None, ne
     w = xw / pw * pixel
     h = yh / ph * pixel
     for i in range(cond.discard, len(cond)):
-        if cond[i] > 0.:
+        if (not isnan(cond[i])) and cond[i] > 0. and (not isinf(cond[i])) and (not isnan(price[i])) and (not isinf(price[i])):
+            print(i, i-w, i+w, price[i]-h, price[i]+h)
             axes.imshow(image, extent=[i-w, i+w, price[i]-h, price[i]+h], *args, **kwargs)
 
     axes.set_aspect('auto')