Matplotlib添加特定刻度表示轴最大值 - 多尺度单一观察

5 投票
1 回答
1851 浏览
提问于 2025-04-17 18:23

我在尝试为每个观察值绘制多个不同的刻度时,成功制作了下面这个图:

enter image description here

不过,我想在每个刻度上添加一个表示y最大值的刻度线,不管它和前一个刻度之间的间隔有多大。下面是一个这样的图例。当y最大值是刻度间隔的倍数时,就会出现这种情况。

enter image description here

谢谢,
F.

下面是用来生成这些示例的代码。

import numpy as np
import pylab as pl
import matplotlib as plt
import matplotlib.ticker as ticker
import matplotlib.transforms

def add_scales(fig, axes, scales, subplot_reduction_factor=0.1, margin_size=50):
    nb_scales = len(scales)
    b,l,w,h = zoom_ax.get_position().bounds

    _, ymax = axes.get_ylim()

    # Saves some space to the right so that we can add our scales
    fig.subplots_adjust(right=1-(subplot_reduction_factor)*nb_scales)

    for (n, (vmin, vmax, color, label, alignment)) in enumerate(scales):

        # Adjust wrt. the orignial figure's scale 
        nax = fig_zoom.add_axes((b,l,w,(h * alignment) / ymax))
        nax.spines['right'].set_position(('outward', -40+n*margin_size))
        nax.set_ylim((vmin,vmax))

        # Move ticks and label to the right
        nax.yaxis.set_label_position('right')
        nax.yaxis.set_ticks_position('right')

        # Hides everything except yaxis
        nax.patch.set_visible(False)
        nax.xaxis.set_visible(False)
        nax.yaxis.set_visible(True)
        nax.spines["top"].set_visible(False)
        nax.spines["bottom"].set_visible(False)

        # Color stuff
        nax.spines['right'].set_color(color)
        nax.tick_params(axis='y', colors=color)
        nax.yaxis.set_smart_bounds(False)
        #nax.yaxis.label.set_color(color)

        if label != None:
            nax.set_ylabel(None)

if __name__ == '__main__':

    a=(np.random.normal(10,5,100))

    a=np.linspace(0,100,100) 
    c=np.linspace(0,80, 100)
    d=np.linspace(0,40,100)


    fig_zoom = plt.pyplot.figure()
    zoom_ax = fig_zoom.add_subplot(1,1,1)


    zoom_ax.plot(a,c)
    zoom_ax.plot(a,d)
    zoom_ax.set_title('Zoom')
    zoom_ax.set_xlabel('A')
    zoom_ax.set_ylabel('B')
    zoom_ax.set_ylim((0,100))
    zoom_ax.grid()
    add_scales(fig_zoom, 
               zoom_ax, [(0,.55,'green',None,40),
                          (0,.85,'blue',None,80)])

    fig_zoom.savefig(open('./test.svg','w'),format='svg')

1 个回答

4

你可以把最高的y轴刻度值设置为你的最大值。如果第二高的y轴刻度值和你的最大值非常接近,标签可能会显得拥挤。

试着在你的循环中添加这个:

tcks = nax.get_yticks()
tcks[-1] = vmax
nax.set_yticks(tcks)

在这里输入图片描述

撰写回答