plt add axis name

1from matplotlib import pyplot as plt    
2
3fig = plt.figure()
4plt.plot(data)
5fig.suptitle('test title', fontsize=20)
6plt.xlabel('xlabel', fontsize=18)
7plt.ylabel('ylabel', fontsize=16)
8fig.savefig('test.jpg')