Merge pull request #14068 from timhoffm/fix-funcantimation-frames-iter-repeat

dopplershift · web-flow · commit 56581cb652b0 · 2019-05-22T15:21:42.000-06:00
Alternative fix for passing iterator as frames to FuncAnimation
diff --git a/lib/matplotlib/animation.py b/lib/matplotlib/animation.py
@@ -1656,7 +1656,14 @@ def __init__(self, fig, func, frames=None, init_func=None, fargs=None,
         elif callable(frames):
             self._iter_gen = frames
         elif np.iterable(frames):
-            self._iter_gen = lambda: iter(frames)
+            if kwargs.get('repeat', True):
+                def iter_frames(frames=frames):
+                    while True:
+                        this, frames = itertools.tee(frames, 2)
+                        yield from this
+                self._iter_gen = iter_frames
+            else:
+                self._iter_gen = lambda: iter(frames)
             if hasattr(frames, '__len__'):
                 self.save_count = len(frames)
         else:
