matplotlib 操作子图（subplot，axes）

Matplotlib 中文用户指南 3.3 使用 GridSpec 自定义子图位置

ax：matplotlib.axes._subplots.AxesSubplot，的基本操作
- ax.set_xticks([]), ax_set_yticks([])：关闭坐标刻度
- ax.axis('off')：关闭坐标轴
- ax.set_title()：设置标题

1. subplots

fig, ax = plt.subplots(nrows=1, ncols=2, figsize=(8, 4))
ax[0].plot(...)
ax[0].set_xlabel(...)
ax[0].set_title(...)

ax[1].plot(...)
ax[1].set_xlabel(...)
...

fig, (ax1, ax2, ax3) = plt.subplots(3, 1, sharex=True)

更一般的做法（也是matlab的风格）是这样的：

fig = plt.figure()          # 创建一个figure对象，底下的一切显示均在此figure上完成

plt.subplot(121)
plt.imshow(img1)
plt.subplot(122)
plt.imshow(img2)

plt.show()

# 当然放在一个range里边
for i in range(ndim):
    plt.subplot(ndim//5, 5, i+1)
    plt.hist()

2. subplot2grid

subplot2grid


# 以下两条语句等价

ax = plt.subplot2grid((2,2),(0, 0))
ax = plt.subplot(221)       # 下标从 0 开始

colspan、rowspan 与索引的对应关系

import matplotlib.pyplot as plt

def disable_axis(ax):
    ax.set_xticks([])
    ax.set_yticks([])
    # ax.axis('off')

def set_title(ax, title):
    ax.set_title(title)

def subplot2grid_demo():
    ax1 = plt.subplot2grid((3, 3), (0, 0), colspan=2)
    disable_axis(ax1)
    set_title(ax1, 'ax1')
    ax2 = plt.subplot2grid((3, 3), (1, 0))
    disable_axis(ax2)
    set_title(ax2, 'ax2')
    ax3 = plt.subplot2grid((3, 3), (1, 1))
    disable_axis(ax3)
    set_title(ax3, 'ax3')
    ax4 = plt.subplot2grid((3, 3), (0, 2), rowspan=2)
    disable_axis(ax4)
    set_title(ax4, 'ax4')
    ax5 = plt.subplot2grid((3, 3), (2, 0), colspan=3)
    disable_axis(ax5)
    set_title(ax5, 'ax5')
    # plt.xticks([])
    # plt.yticks([])
    # plt.axis('off')
    plt.show()

if __name__ == '__main__':
    subplot2grid_demo()

3. GridSpec：方便的切片操作

GridSpec 提供了十分方便的切片操作，实现上述功能，则只需如下代码：

def gridspec_demo():
    gs = gridspec.GridSpec(3, 3)
    ax1 = plt.subplot(gs[0, :2])
    ax2 = plt.subplot(gs[1, 0])
    ax3 = plt.subplot(gs[1, 1])
    ax4 = plt.subplot(gs[0:2, 2])
    ax5 = plt.subplot(gs[2:, :])
    plt.show()