Thanks @kgoodrick, your answer works perfectly. Just one addition: I had to change axis(False) to axis(None) for it to work. If you post it as an answer (not a comment), then I can click the "accept" button :-)