使用host_subplot时如何创建不同大小的子图?

时间:2017-04-17 01:49:25

标签: python matplotlib plot

这是我previous question的后续行动。该问题的答案显示了第二次使用host_subplot生成第二个图表的解决方案,为方便起见,下面将其复制。

我尝试使用this answer,但在尝试将Unknown property gridspec_kw传递给gridspec_kw函数时收到了错误host_subplot

当其中一个图表使用host_subplot的双Y轴时,如何获得不同大小的子图?

"""
Parasite axis demo

The following code is an example of a parasite axis. It aims to show a user how
to plot multiple different values onto one single plot. Notice how in this
example, par1 and par2 are both calling twinx meaning both are tied directly to
the x-axis. From there, each of those two axis can behave separately from the
each other, meaning they can take on separate values from themselves as well as
the x-axis.
"""
from mpl_toolkits.axes_grid1 import host_subplot
import mpl_toolkits.axisartist as AA
import matplotlib.pyplot as plt

host = host_subplot(211, axes_class=AA.Axes)
plt.subplots_adjust(right=0.75)

par1 = host.twinx()
par2 = host.twinx()

offset = 60
new_fixed_axis = par2.get_grid_helper().new_fixed_axis
par2.axis["right"] = new_fixed_axis(loc="right",
                                    axes=par2,
                                    offset=(offset, 0))

par2.axis["right"].toggle(all=True)

host.set_xlim(0, 2)
host.set_ylim(0, 2)

host.set_xlabel("Distance")
host.set_ylabel("Density")
par1.set_ylabel("Temperature")
par2.set_ylabel("Velocity")

p1, = host.plot([0, 1, 2], [0, 1, 2], label="Density")
p2, = par1.plot([0, 1, 2], [0, 3, 2], label="Temperature")
p3, = par2.plot([0, 1, 2], [50, 30, 15], label="Velocity")

par1.set_ylim(0, 4)
par2.set_ylim(1, 65)

host.legend()

host.axis["left"].label.set_color(p1.get_color())
par1.axis["right"].label.set_color(p2.get_color())
par2.axis["right"].label.set_color(p3.get_color())

## How can I make this subplot only 1/3 as tall as the first one?
host2 = host_subplot(212, axes_class=AA.Axes, sharex=host)
host2.plot([0,1,2], [5,6,7])
host2.set_title('Sharing X axis')

plt.draw()
plt.show()

2 个答案:

答案 0 :(得分:1)

通过将host_subplot的第一个参数替换为GridSpec,似乎可以使用accepted answer对链接的问题进行此操作。添加和更改的行在下面。

from matplotlib import gridspec

# Two rows, one column, with height ratio of 4:1 between the top and bottom graph.
gs = gridspec.GridSpec(2, 1, height_ratios=[4, 1])

host = host_subplot(gs[0], axes_class=AA.Axes)
host2 = host_subplot(gs[1], axes_class=AA.Axes, sharex=host)

答案 1 :(得分:0)

您可以通过将两次调用中的行数更改为host_subplot()来实现此目的。对于3:1的高度差:

host = host_subplot(211, axes_class=AA.Axes)
host2 = host_subplot(614, axes_class=AA.Axes, sharex=host)

第一个调用将图形分为2行,并选择上半部分的轴。第二个调用将数字视为分为6行并选择第4行中的子图。

H / T:https://stackoverflow.com/a/3339129/7517724