Hinton diagrams in matplotlib