如何对两个以上的x比例子图和两个以上的y比例子图进行多个子图绘制?

时间:2018-09-29 01:27:47

标签: python-3.x matplotlib figure subplot multiple-axes

我正在尝试制作一个2x2子图,每个内部子图都由两个x轴和两个y轴组成;第一个xy对应于线性刻度,第二个xy对应于对数刻度。在假设之前已经问过这个问题之前,请先咨询matplotlib docs和examples show how to do multiple scales for either xy,但不要同时问两者。 This post on stackoverflow是最接近我的问题的东西,我试图用这个想法来实现我想要的东西。我的尝试在下面。

首先,我们初始化数据,滴答和滴答标签。想法是,备用刻度将具有相同的刻度位置,并且刻度标签已更改,以反映备用刻度。

import numpy as np
import matplotlib.pyplot as plt

# xy data (global)
X = np.linspace(5, 13, 9, dtype=int)
Y = np.linspace(7, 12, 9)

# xy ticks for linear scale (global)
dtick = dict(X=X, Y=np.linspace(7, 12, 6, dtype=int))

# xy ticklabels for linear and logarithmic scales (global)
init_xt = 2**dtick['X']
dticklabel = dict(X1=dtick['X'], Y1=dtick['Y']) # linear scale
dticklabel['X2'] = ['{}'.format(init_xt[idx]) if idx % 2 == 0 else '' for idx in range(len(init_xt))] # log_2 scale
dticklabel['Y2'] = 2**dticklabel['Y1'] # log_2 scale

借用链接的SO帖子,我将在4个子图中的每个图中绘制相同的内容。由于在每个子图中两个缩放比例都使用了类似的方法,因此该方法被丢入for循环中。但是我们需要每个行号,列号和图号。

# 2x2 subplot
# fig.add_subplot(row, col, pnum); corresponding iterables = (irows, icols, iplts)
irows = (1, 1, 2, 2)
icols = (1, 2, 1, 2)
iplts = (1, 2, 1, 2)
ncolors = ('red', 'blue', 'green', 'black')

将所有这些放在一起,输出图的功能如下:

def initialize_figure(irows, icols, iplts, ncolors, figsize=None):
    """ """
    fig = plt.figure(figsize=figsize)
    for row, col, pnum, color in zip(irows, icols, iplts, ncolors):
        ax1 = fig.add_subplot(row, col, pnum) # linear scale
        ax2 = fig.add_subplot(row, col, pnum, frame_on=False) # logarithmic scale ticklabels
        ax1.plot(X, Y, '-', color=color)
        # ticks in same positions
        for ax in (ax1, ax2):
            ax.set_xticks(dtick['X'])
            ax.set_yticks(dtick['Y'])
        # remove xaxis xtick_labels and labels from top row
        if row == 1:
            ax1.set_xticklabels([])
            ax2.set_xticklabels(dticklabel['X2'])
            ax1.set_xlabel('')
            ax2.set_xlabel('X2', color='gray')
        # initialize xaxis xtick_labels and labels for bottom row
        else:
            ax1.set_xticklabels(dticklabel['X1'])
            ax2.set_xticklabels([])
            ax1.set_xlabel('X1', color='black')
            ax2.set_xlabel('')
        # linear scale on left
        if col == 1:
            ax1.set_yticklabels(dticklabel['Y1'])
            ax1.set_ylabel('Y1', color='black')
            ax2.set_yticklabels([])
            ax2.set_ylabel('')
        # logarithmic scale on right
        else:
            ax1.set_yticklabels([])
            ax1.set_ylabel('')
            ax2.set_yticklabels(dticklabel['Y2'])
            ax2.set_ylabel('Y2', color='black')

        ax1.tick_params(axis='x', colors='black')
        ax1.tick_params(axis='y', colors='black')
        ax2.tick_params(axis='x', colors='gray')
        ax2.tick_params(axis='y', colors='gray')
        ax1.xaxis.tick_bottom()
        ax1.yaxis.tick_left()
        ax1.xaxis.set_label_position('top')
        ax1.yaxis.set_label_position('right')
        ax2.xaxis.tick_top()
        ax2.yaxis.tick_right()
        ax2.xaxis.set_label_position('top')
        ax2.yaxis.set_label_position('right')
        for ax in (ax1, ax2):
            ax.set_xlim([4, 14])
            ax.set_ylim([6, 13])
    fig.tight_layout()
    plt.show()
    plt.close(fig)

调用initialize_figure(irows, icols, iplts, ncolors)会产生下图。

enter image description here

我正在应用相同的xlimylim,所以我不明白为什么子图的大小都不同。另外,轴标签和轴刻度标签不在指定位置(因为fig.add_subplot(...)的索引从1开始而不是从0开始。

我的错误是什么,如何获得预期的结果?

(如果不清楚,我尝试将xticklabelsxlabels的线性比例尺的xticklabelsxlabels放在底部对于最上面一行的对数标度,右列右侧的对数标度为'yticklabels and ylabels for the linear scale on the left side of the left column, and the 'yticklabelsylabelscolor='black' kwarg对应线性标度,color='gray' kwarg对应对数标度。)

1 个答案:

答案 0 :(得分:1)

该代码中的<Field component={MyCustomizedDropdown} /> irows列表没有任何作用。要在2x2网格中创建4个子图,您需要在icols上进行遍历,

range(1,5)

这可能不是代码中的唯一问题,但是只要未正确创建子图,就不值得进一步研究。