rtgym.utils.visualization package
Submodules
rtgym.utils.visualization.plot_ratemap module
- plot_ratemaps(ratemaps, n_rows, n_cols, cmap='jet', per_image_height=2, arena_map=None)[source]
Plots a grid of average rate maps with specified rows and columns, dynamically adjusting the figure size to maintain aspect ratios.
- Parameters:
ratemaps – numpy array of shape (n, w, h)
n_rows – number of rows in the subplot grid
n_cols – number of columns in the subplot grid
cmap – colormap to use for the images
per_image_height – height of each individual subplot in inches (default: 2)