Use matplotlib for plotting