Matplotlib apply colormap to array.