Skip to content

Commit 5639974

Browse files
committed
BUG: Fix streamplot when velocity component is exactly zero.
1 parent 2c91aa6 commit 5639974

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

lib/matplotlib/streamplot.py

+5-5
Original file line numberDiff line numberDiff line change
@@ -461,14 +461,14 @@ def _euler_step(xf_traj, yf_traj, dmap, f):
461461
xi = xf_traj[-1]
462462
yi = yf_traj[-1]
463463
cx, cy = f(xi, yi)
464-
if cx > 0:
465-
dsx = (nx - 1 - xi) / cx
466-
else:
464+
if cx < 0:
467465
dsx = xi / -cx
468-
if cy > 0:
469-
dsy = (ny - 1 - yi) / cy
470466
else:
467+
dsx = (nx - 1 - xi) / cx
468+
if cy < 0:
471469
dsy = yi / -cy
470+
else:
471+
dsy = (ny - 1 - yi) / cy
472472
ds = min(dsx, dsy)
473473
xf_traj.append(xi + cx*ds)
474474
yf_traj.append(yi + cy*ds)

0 commit comments

Comments
 (0)