Matplotlib 简明教程

Matplotlib - Subplots() Function

Matplotlib 的 pyplot API 有一个名为 subplots() 的便捷函数，它充当一个实用程序包装器，并有助于在一个调用中创建子图的常用布局，包括包围图形对象。

Plt.subplots(nrows, ncols)

此函数的两个整型参数指定子图网格的行数和列数。该函数返回一个图形对象和一个包含与 nrows*ncols 相等的轴对象元组。每个轴对象都可以通过其索引进行访问。在这里，我们创建了一个 2 行 2 列的子图，并在每个子图中显示 4 个不同的绘图。

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

上述代码行生成以下输出 -

subplots function