fix N**2 memory complexity bug

Author: theRealSuperMario

test/test_cleanfigure.py

@@ -1047,3 +1047,12 @@ def test_getHeightWidthInPixels():
         assert w == 600 and h == 400
         w, h = cleanfigure._get_width_height_in_pixels(fig, 600)
         assert w == h
+
+
+def test_memory():
+    import matplotlib.pyplot as plt
+    import numpy as np
+    import tikzplotlib
+
+    plt.plot(np.arange(100000))
+    tikzplotlib.clean_figure()

tikzplotlib/_cleanfigure.py

@@ -949,7 +949,7 @@ def _segment_visible(data, dataIsInBox, xLim, yLim):
     :type yLim: list, np.array
     """
     n = np.shape(data)[0]
-    mask = np.zeros((n - 1, n)) == 1
+    mask = np.zeros((n - 1, 1)) == 1
 
     # Only check if there is more than 1 point
     if n > 1: