from matplotlib.pyplot import figure code example

Example: import pyplot python

import matplotlib.pyplot as plt