Save Figure

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(0, 1, 50)
plt.plot(x, x ** 2)

plt.title("Saved figure")

# Raster formats
plt.savefig("out.png", dpi=150, bbox_inches="tight", facecolor="white")
plt.savefig("out.jpg", dpi=120)

# Vector formats (scalable, good for print)
plt.savefig("out.pdf")
plt.savefig("out.svg")

# Transparent background
plt.savefig("overlay.png", transparent=True)

# Save the current figure as an array
buf = np.asarray(plt.gcf().canvas.buffer_rgba())
print(buf.shape)