Multi-Subplot Image Update

Example showing updating a multiple subplots with new random 512x512 data.

# test_example = false

import fastplotlib as fpl
import numpy as np

# Figure of shape 2 x 3 with all controllers synced
figure = fpl.Figure(shape=(2, 3), controller_ids="sync", size=(700, 560))

# Make a random image graphic for each subplot
for subplot in figure:
    # create image data
    data = np.random.rand(512, 512)
    # add an image to the subplot
    subplot.add_image(data, name="rand-img")

figure[0,1]["rand-img"].cmap = "viridis"
figure[1,0]["rand-img"].cmap = "Wistia"
figure[0,2]["rand-img"].cmap = "gray"
figure[1,1]["rand-img"].cmap = "spring"

# Define a function to update the image graphics with new data
# add_animations will pass the gridplot to the animation function
def update_data(f):
    for subplot in f:
        new_data = np.random.rand(512, 512)
        # index the image graphic by name and set the data
        subplot["rand-img"].data = new_data

# add the animation function
figure.add_animations(update_data)

# show the gridplot
figure.show()


# NOTE: `if __name__ == "__main__"` is NOT how to use fastplotlib interactively
# please see our docs for using fastplotlib interactively in ipython and jupyter
if __name__ == "__main__":
    print(__doc__)
    fpl.loop.run()