how to remove axis in matplotlib code example

Example 1: pyplot not show axis

plt.axis('off')

Example 2: plt off axis

from numpy import random
import matplotlib.pyplot as plt

data = random.random((5,5))
img = plt.imshow(data, interpolation='nearest')
img.set_cmap('hot')
plt.axis('off')
plt.savefig("test.png", bbox_inches='tight')