matplotlib 子图

subplots()

帮助在单个调用中创建子图的通用布局，包括封闭的图形对象。

plt.subplots(nrows, ncols)

其中的nrows指定子图的行数，ncols指定子图的列数。

import matplotlib.pyplot as plt
import numpy as np

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

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.exp(x))
a[1][0].set_title('exp')
a[1][1].plot(x,np.log10(x))
a[1][1].set_title('log')
plt.show()

注：a[0][0]指第一个子图， a[0][1] 指横向第二个子图，依此类推。