使用pyplot的subplot2grid()函数修改子图位置

使用pyplot的subplot()函数可以在规划好的某个区域中绘制单个子图，subplot()函数的语法格式如下：

subplot(nrows,  ncols,  index,  projection,  polar,  sharex,  sharey, label, **kwargs)

该函数的常用参数如下：

参数	描述
nrows	表示规划区域的行数
ncols	表示规划区域的列数
index	表示选择区域的索引，默认从1开始编号
projection	表示子图的投影类型
polar	表示是否使用极坐标
sharex	表示是否共享子图x轴或y轴

原代码及原图如下：

import numpy as np
import matplotlib.pyplot as plt

plt.rcParams['font.sans-serif'] = ["SimHei"]

x = [x for x in range(1, 13)]
y1 = [20, 28, 23, 16, 29, 36, 39, 33, 31, 19, 21, 25]
y2 = [17, 22, 39, 26, 35, 23, 25, 27, 29, 38, 28, 20]
labels = ['1 月', '2 月', '3 月', '4 月', '5 月', '6 月', '7月', '8 月', '9 月', '10 月', '11 月', '12 月']

# 绘制第一个子图
ax1 = plt.subplot(211)
ax1.plot(x, y1, 'm--o', lw=2, ms=5, label='产品A')
ax1.plot(x, y2, 'g--o', lw=2, ms=5, label='产品B')
ax1.set_title("产品A 与产品B的销售额", fontsize=11)
ax1.set_ylim(10, 45)
ax1.set_ylabel('销售额(亿元)')
ax1.set_xlabel('月份')
for xy1 in zip(x, y1):
    ax1.annotate("%s" % xy1[1], xy=xy1, xytext=(-5, 5), textcoords='offset points')
for xy2 in zip(x, y2):
    ax1.annotate("%s" % xy2[1], xy=xy2, xytext=(-5, 5), textcoords='offset points')
ax1.legend()

# 绘制第二个子图
ax2 = plt.subplot(223)
ax2.pie(y1, radius=1, wedgeprops={'width':0.5}, labels=labels, autopct='%3.1f%%', pctdistance=0.75)
ax2.set_title('产品A的销售额 ')

# 绘制第三个子图
ax3 = plt.subplot(224)
ax3.pie(y2, radius=1, wedgeprops={'width':0.5}, labels=labels,autopct='%3.1f%%', pctdistance=0.75)
ax3.set_title('产品B的销售额 ')

# 调整子图之间的距离
plt.tight_layout()
plt.show()

如图所示：ax1在画布被规划成为 2 X 1 的矩阵区域，之后在索引为1的区域中绘图，而ax2和ax3则在画布被规划成为 2 X 2 的矩阵区域，之后分别在索引3和4区域中绘制子图

---

学习目标：

一、利用subplot2grid()修改下列代码，使其运行结果如图所示（2行4列)

pyplot的subplot2grid()函数可以将整个画布规划成非等分布局的区域，并可在选中的某个区域中绘制子图，subplot2grid()函数的语法格式如下：

subplot2grid(shape, loc, rowspan=1, colspan=1, fig=None, **kwargs)

该函数的常用参数如下：

参数	描述
shape	要在其中放置轴的网格的行数和列数
loc	轴在网格中的位置的行号和列号
rowspan	轴向下跨度的行数
colspan	轴向右跨度的列数
fig	要放置子图的图。默认为当前数字

修改后的代码如下：

import numpy as np
import matplotlib.pyplot as plt

plt.rcParams['font.sans-serif'] = ["SimHei"]

x = [x for x in range(1, 13)]
y1 = [20, 28, 23, 16, 29, 36, 39, 33, 31, 19, 21, 25]
y2 = [17, 22, 39, 26, 35, 23, 25, 27, 29, 38, 28, 20]
labels = ['1 月', '2 月', '3 月', '4 月', '5 月', '6 月', '7月', '8 月', '9 月', '10 月', '11 月', '12 月']

# 绘制第一个子图
ax1 = plt.subplot2grid((2, 4), (0, 1), colspan = 2)  
# 画布被规划成 2 X 4 的矩阵区域，之后在第0行第1列的区域中绘制一个向右跨越2列的子图
ax1.plot(x, y1, 'm--o', lw=2, ms=5, label='产品A')
ax1.plot(x, y2, 'g--o', lw=2, ms=5, label='产品B')
ax1.set_title("产品A 与产品B的销售额", fontsize=11)
ax1.set_ylim(10, 45)
ax1.set_ylabel('销售额(亿元)')
ax1.set_xlabel('月份')
for xy1 in zip(x, y1):
    ax1.annotate("%s" % xy1[1], xy=xy1, xytext=(-5, 5), textcoords='offset points')
for xy2 in zip(x, y2):
    ax1.annotate("%s" % xy2[1], xy=xy2, xytext=(-5, 5), textcoords='offset points')
ax1.legend()

# 绘制第二个子图
ax2 = plt.subplot2grid((2, 4), (1, 0))
# 画布被规划成 2 X 4 的矩阵区域，之后在第1行第0列的区域中绘制子图
ax2.pie(y1, radius=1, wedgeprops={'width':0.5}, labels=labels, autopct='%3.1f%%', pctdistance=0.75)
ax2.set_title('产品A的销售额 ')

# 绘制第三个子图
ax3 = plt.subplot2grid((2, 4), (1, 3))
# 画布被规划成 2 X 4 的矩阵区域，之后在第1行第3列的区域中绘制子图
ax3.pie(y2, radius=1, wedgeprops={'width':0.5}, labels=labels,autopct='%3.1f%%', pctdistance=0.75)
ax3.set_title('产品B的销售额 ')

# 调整子图之间的距离
plt.tight_layout()
plt.show()

修改后最终运行结果与上图一致：