You'll probably have to make a custom colormap and then pass that right in to matrix_plot, perhaps using ListedColormap or LinearSegmentedColormap. But I think this should be doable, as we use matplotlib colormaps as they stand.