from matplotlib import pyplot as plt # As png plt.savefig('Path/FigureName.png') # As pdf plt.savefig('Path/FigureName.pdf')
import imageio im = imageio.imread('my_image.png') print(im.shape)