matplotlib：subplot——子图绘制

在matplotlib中一个figure可以包含多个子图，使用subplot可以实现，先看下面的实例：

plt.figure(1)
plt.subplot(2, 2, 1)
plt.subplot(2, 2, 2)
plt.subplot(2, 1, 2)
plt.show()

效果如下：

subplot有三个参数： numRows，numCols，PlotNum

numRows代表行数，numCols代表列数，整个绘图区域分为numRows*numCols的子区域

PlotNum代表子图的编号，按照从上到下，从左到右的顺序编号

 subplot的参数可以用 ‘，’隔开，也可以不用

plt.subplot(2, 2, 1)
plt.subplot(221)

现在我们来实现一个小程序，接受两个输入，第一个输入代表绘图区域被分为多少行，第二个输入代表每一行有多少子图。

例如，第一个输入r=4，代表划分为4行，第二个输入l="1，2，3，4"是一个字符串，代表每一行的子图数，从而规划一个绘图布局，实现的效果如下：

 实现代码如下：

import matplotlib.pyplot as plt
import numpy as np

def figure_layout(r:int, l:str):
    n = [int(i) for i in l.split(',')]
    ax = []
    fig = plt.figure(1)
    for i in range(r):
        for j in range(n[i]):
            ax.append(fig.add_subplot(r, n[i], i*n[i]+j+1))
    return fig

if __name__ == "__main__":
    r = 4
    l = "1,2,3,4"
    fig = figure_layout(r, l)
    plt.show()