Error: AttributeError: module 'matplotlib' has no attribute 'figure'

Error: AttributeError: module 'matplotlib' has no attribute 'figure'
Before modification:

import matplotlib as pltplt.figure()plt.imshow(trainImages[10])plt.colorbar()plt.grid(False)

Modified to: import matplotlib.pyplot as plt
After modification:

import matplotlib.pyplot as pltplt.figure()plt.imshow(trainImages[10])plt.colorbar()plt.grid(False)