When using matplotlib.pyplot.savefig
, the file format can be specified by the extension:
from matplotlib import pyplot as plt
plt.savefig('foo.png')
plt.savefig('foo.pdf')
That gives a rasterized or vectorized output respectively.
In addition, there is sometimes undesirable whitespace around the image, which can be removed with:
plt.savefig('foo.png', bbox_inches="tight")
Note that if showing the plot, plt.show()
should follow plt.savefig()
; otherwise, the file image will be blank.