def drain(self): for i,(info, (ys,us,vs)) in enumerate(self.iterframes()): nim = Nimage(Grid(None, shape=ys.shape), data=ys) nim.save_raster(os.path.join(self.outdir, '%04d.png'%self.framei)) self.framei += 1