Skip to content

Commit 84a4bbd

Browse files
Fix non-square wireframe plots
fix rstride, cstride = 0 rc 0
1 parent d08fd8f commit 84a4bbd

File tree

1 file changed

+13
-1
lines changed

1 file changed

+13
-1
lines changed

lib/mpl_toolkits/mplot3d/axes3d.py

Lines changed: 13 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2445,7 +2445,19 @@ def plot_wireframe(self, X, Y, Z, *, axlim_clip=False, **kwargs):
24452445

24462446
row_lines = np.stack([X[rii], Y[rii], Z[rii]], axis=-1)
24472447
col_lines = np.stack([tX[cii], tY[cii], tZ[cii]], axis=-1)
2448-
lines = np.concatenate([row_lines, col_lines])
2448+
2449+
nr, nc = len(rii), len(cii)
2450+
if nr == nc:
2451+
lines = np.concatenate([row_lines, col_lines])
2452+
elif nr == 0:
2453+
lines = col_lines
2454+
elif nc == 0:
2455+
lines = row_lines
2456+
else:
2457+
lines = np.full((nr + nc, max(row_lines.shape[1], col_lines.shape[1]), 3),
2458+
np.nan)
2459+
lines[:nr, :row_lines.shape[1], :] = row_lines
2460+
lines[nr:, :col_lines.shape[1], :] = col_lines
24492461

24502462
linec = art3d.Line3DCollection(lines, axlim_clip=axlim_clip, **kwargs)
24512463
self.add_collection(linec)

0 commit comments

Comments
 (0)