python waterfall plot matplotlib