matplotlib 获取给定图形包含的所有轴以应用一些设置 [英] matplotlib get all axes that a given figure contains to apply some settings

问题描述

我正在编写一个修改图形上的轴尺寸和位置的函数，但是当出现双轴时会出现问题:

I'm writing a function that modifies the axes size and position on a figure, but when comes twin axes it makes a problem:

import matplotlib.pyplot as plt

def fig_layout(fig, vspace = 0.3):  #  function to make space at the bottom for legend box and
                                    #+ other text input
    for ax in ~~~fig.axes~~~: #  Here 'fig.axes' is not right, I need to find the exact syntax
                              #+ I need to put
        box = ax.get_position()
        ax.set_position([box.x0, box.y0 + box.height * vspace,
        box.width, box.height * (1 - vspace)])

x = np.arange(10)
fig = plt.figure()
ax1 = fig.add_subplot(1, 1, 1)
n = 3
line = {}
for i in range(3):
    line['lines'].append(ax1.plot(x, i*x**2))
    line['labels'].append(r'$y = %i \cdot x^2$'%i)
ax1.set_title('example plot')

ax2 = ax1.twinx()
line['lines'].append(ax2.plot(x, x^-1, label = r'$y = x^-1$'))
line['labels'].append(r'$y = x^-1$')
leg = ax1.legend(line['lines'], line['labels'])
fig_layout(fig)
#  I will put the legend box at the bottom of the axes with another function.
plt.show()

推荐答案

我认为您可以使用fig.get_axes().

I think you can use fig.get_axes().

例如，要修改第一个子图的标题，您可以执行以下操作:

For example, to modify the title of the first sub-plot, you can do:

plt.gcf().get_axes()[0].set_title("example plot")

这篇关于matplotlib 获取给定图形包含的所有轴以应用一些设置的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！