diff --git a/lib/mpl_toolkits/mplot3d/axes3d.py b/lib/mpl_toolkits/mplot3d/axes3d.py
index c56e4c6b7039..3122b64bde73 100644
--- a/lib/mpl_toolkits/mplot3d/axes3d.py
+++ b/lib/mpl_toolkits/mplot3d/axes3d.py
@@ -2890,8 +2890,10 @@ def add_collection3d(self, col, zs=0, zdir='z', autolim=True, *,
if autolim:
if isinstance(col, art3d.Line3DCollection):
- self.auto_scale_xyz(*np.array(col._segments3d).transpose(),
- had_data=had_data)
+ # Handle ragged arrays by extracting coordinates separately
+ all_points = np.concatenate(col._segments3d)
+ self.auto_scale_xyz(all_points[:, 0], all_points[:, 1],
+ all_points[:, 2], had_data=had_data)
elif isinstance(col, art3d.Poly3DCollection):
self.auto_scale_xyz(col._faces[..., 0],
col._faces[..., 1],
diff --git a/lib/mpl_toolkits/mplot3d/tests/test_axes3d.py b/lib/mpl_toolkits/mplot3d/tests/test_axes3d.py
index e6d11f793b46..791fa5bb3fbd 100644
--- a/lib/mpl_toolkits/mplot3d/tests/test_axes3d.py
+++ b/lib/mpl_toolkits/mplot3d/tests/test_axes3d.py
@@ -2689,3 +2689,27 @@ def test_ndarray_color_kwargs_value_error():
ax = fig.add_subplot(111, projection='3d')
ax.scatter(1, 0, 0, color=np.array([0, 0, 0, 1]))
fig.canvas.draw()
+
+
+def test_line3dcollection_autolim_ragged():
+ """Test Line3DCollection with autolim=True and lines of different lengths."""
+ fig = plt.figure()
+ ax = fig.add_subplot(projection='3d')
+
+ # Create lines with different numbers of points (ragged arrays)
+ edges = [
+ [(0, 0, 0), (1, 1, 1), (2, 2, 2)], # 3 points
+ [(0, 1, 0), (1, 2, 1)], # 2 points
+ [(1, 0, 1), (2, 1, 2), (3, 2, 3), (4, 3, 4)] # 4 points
+ ]
+
+ # This should not raise an exception with the fix
+ collections = ax.add_collection3d(art3d.Line3DCollection(edges), autolim=True)
+
+ # Check that limits were computed correctly with margins
+ # The limits should include all points with default margins
+ assert np.allclose(ax.get_xlim3d(), (-0.08333333333333333, 4.083333333333333))
+ assert np.allclose(ax.get_ylim3d(), (-0.0625, 3.0625))
+ assert np.allclose(ax.get_zlim3d(), (-0.08333333333333333, 4.083333333333333))
+
+ plt.close(fig)
pFad - Phonifier reborn
Pfad - The Proxy pFad of © 2024 Garber Painting. All rights reserved.
Note: This service is not intended for secure transactions such as banking, social media, email, or purchasing. Use at your own risk. We assume no liability whatsoever for broken pages.
Alternative Proxies:
Alternative Proxy
pFad Proxy
pFad v3 Proxy
pFad v4 Proxy