I have a python plot in 3 dimensions which I am trying to export as a flat image where the z height is encoded in the colour. I think its to do with the pcolor function but