fixed_plot.py

import matplotlib.pyplot as plt
import matplotlib.animation as animation
import numpy as np
from PIL import Image

def create_animation():
    # Load the same data
    file = r'example.dat'
    nx = 8
    ny = 9
    num_frames = 72
    data = np.fromfile(file, np.float32).reshape(num_frames, ny, nx)
    
    # Create figure and plot
    fig, ax = plt.subplots()
    
    # Use exactly the same parameters as bugReproduction4.py
    vmax = 100
    vmin = 0
    
    # Create plot exactly like bugReproduction4.py
    h = ax.imshow(data[0], cmap=plt.get_cmap('CMRmap_r'), origin='lower', 
                  interpolation='none', vmin=vmin, vmax=vmax, animated=True)
    
    # Set the same axis labels
    ax.set_xticks(range(nx))
    ax.set_xticklabels(range(1, nx + 1))
    ax.set_yticks(range(ny))
    ax.set_yticklabels(range(1, ny + 1))
    
    plt.colorbar(h)
    fig.tight_layout()

    def update(frame):
        img = data[frame,]
        h.set_array(img)
        return h,

    # Create animation with same parameters
    interval = 100
    ani = animation.FuncAnimation(fig, update, frames=range(num_frames),
                                interval=interval, blit=True)
    
    # Save first, then show - exactly like bugReproduction4.py
    ani.save('example_fixed.gif', writer='pillow', fps=2, dpi=300)
    plt.show()

def display_saved_gif():
    # Load the gif using PIL
    gif = Image.open('example_fixed.gif')
    
    # Create figure and axis
    fig, ax = plt.subplots()
    
    def update(frame):
        # Seek to frame
        gif.seek(frame)
        # Convert PIL image to array and display
        ax.clear()
        ax.imshow(gif)
        return ax,
    
    # Get number of frames
    n_frames = 0
    try:
        while True:
            gif.seek(n_frames)
            n_frames += 1
    except EOFError:
        gif.seek(0)
    
    # Create animation
    ani = animation.FuncAnimation(fig, update, frames=range(n_frames),
                                interval=500, blit=True)
    plt.show()

if __name__ == '__main__':
    create_animation()
    print("\nNow displaying the saved gif:")
    display_saved_gif()