Use matplotlib's stem function. And PEP8.