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()

巴特西

matplotlib 操作子图（subplot，axes）

1. subplots

2. subplot2grid

3. GridSpec：方便的切片操作

最新文章

热门文章