用subplots()函数绘制子图

matplotlib.pyplot模块提供了一个 subplots() 函数

subplots 的函数如下：

fig , ax = plt.subplots(nrows, ncols)

nrows 与 ncols 表示两个整数参数，它们指定子图所占的行数、列数。
该函数返回值是一个元组，包括一个图形对象和所有的 axes 对象。其中 axes 对象的数量等于 nrows * ncols，axes 对象可通过索引值来访问。

import matplotlib.pyplot as plt

fig,a = plt.subplots(2,2)

import numpy as np

x = np.arange(1,5)


#绘制平方函数

a[0][0].plot(x,x*x)

a[0][0].set_title('square')


#绘制平方根图像

a[0][1].plot(x,np.sqrt(x))

a[0][1].set_title('square root')



#绘制对数函数

a[1][0].plot(x,np.log10(x))

a[1][0].set_title('log')




#绘制指数函数

a[1][1].plot(x,np.exp(x))

a[1][1].set_title('exp')

plt.show()