使用Matplotlib绘制打断图Broken Axis

使用Matplotlib绘制打断坐标轴Broken Axis

对于一批存在离群点的或者两极分化的数据，为了突出其值域差异，时常需要用到打断坐标轴效果。

使用Matplotlib绘制的效果如下：

对于同样的数据，使用brokenaxes库的绘制效果如下：

绘制Broken Axis的关键是需要用户确定在哪里打断坐标轴。下面是两种绘制方式的代码示例：

仅使用matplotlib

import matplotlib.pyplot as plt
import numpy as np
from brokenaxes import brokenaxes

BROKEN_SPACE = 0.05#打断部分的长度

np.random.seed(19680801)#使用相同的随机数种子以便得到相同的数据

pts = np.random.rand(30)*.2
# 将序号为3、14的两个数据的y坐标加入噪声，使它们成为离群点
pts[[3, 14]] += .8

if __name__=="__main__":
    # ax1 用于显示离群点
    # ax2 用于显示其他点
    fig, (ax1, ax2) = plt.subplots(2, 1, sharex=True)
    fig.subplots_adjust(hspace=BROKEN_SPACE)  # adjust space between axes

    # 在ax1、ax2上绘制同一批数据
    ax1.plot(pts)
    ax2.plot(pts)

    # 用户确定打断位置后，分别设置ax1、ax2上要显示的数据范围（在这个例子中是y坐标范围）
    ax1.set_ylim(.78, 1.)  # 离群点范围
    ax2.set_ylim(0, .22)  # 其他数据范围

    # 隐藏ax1、ax2之间的实线、
    ax1.spines.bottom.set_visible(False)
    ax2.spines.top.set_visible(False)
    # 隐藏刻度线
    ax1.xaxis.tick_top()
    ax1.tick_params(labeltop=False)  # don't put tick labels at the top
    ax2.xaxis.tick_bottom()

    # 绘制中断斜线
    d = .5  # 斜线的y方向范围，值越大斜线越接近竖直
    kwargs = dict(marker=[(-1, -d), (1, d)], markersize=12,
                linestyle="none", color='k', mec='k', mew=1, clip_on=False)
    ax1.plot([0, 1], [0, 0], transform=ax1.transAxes, **kwargs)
    ax2.plot([0, 1], [1, 1], transform=ax2.transAxes, **kwargs)


    plt.show()

使用matplotlib+brokenaxes库绘制

安装brokenaxes库

使用pip安装：

pip install brokenaxes

安装后，即可在python文件中引入brokenaxes模块。

import matplotlib.pyplot as plt
import numpy as np
from brokenaxes import brokenaxes

np.random.seed(19680801)

pts = np.random.rand(30)*.2
# 将序号为3、14的两个数据的y坐标加入噪声，使它们成为离群点
pts[[3, 14]] += .8

if __name__=="__main__":
    # (0, .22), (.78, 1)表示在0.22-0.78之间打断
    bax = brokenaxes(ylims=((0, .22), (.78, 1)), hspace=.09)
    bax.plot(pts, label='data')
    bax.legend(loc=3)
    bax.set_xlabel('x')
    bax.set_ylabel('value')
    plt.show()