Rearrange elements in numpy array to plot 3d points in 2d, preserving grouping

Question

Hello I have a 3d numpy array of shape (nc, nx, ny). I would like to plot it (with matpoltlib) using x and y as the axis, but in such a way that all c values are close to each o