在yt-Project图中添加子图

Question

我正在使用yt-Project库来可视化数据并创建绘图。 现在，我想创建一个包含两个子图的图。看来这不是yt直接可能的，你必须使用matplotlib进行进一步的自定义（描述here）。 不习惯matplotlib（以及一般的python）我试过这样的事情：

slc = yt.SlicePlot(ds, 'x', 'density')
dens_plot = slc.plots['density']

fig = dens_plot.figure
ax = dens_plot.axes
#colorbar_axes = dens_plot.cax

new_ax2 = fig.add_subplot(212)

slc.save()

但不是在第一个下面添加另一个子图，而是将其添加到其中。

我想要实现的是来自不同数据集的另一个图，其中相同的颜色条和第一个正下方的x和y轴相同。

感谢您的帮助。

Answer 1

现在，最简单的方法是使用AxesGrid，in this yt cookbook example以及this one。

这是一个使用yt 3.2.1绘制时间序列中两次气体密度的示例。我可以从http://yt-project.org/data下载我使用的示例数据。

import yt
import matplotlib.pyplot as plt
from mpl_toolkits.axes_grid1 import AxesGrid

fns = ['enzo_tiny_cosmology/DD0005/DD0005', 'enzo_tiny_cosmology/DD0040/DD0040']

fig = plt.figure()

# See http://matplotlib.org/mpl_toolkits/axes_grid/api/axes_grid_api.html
# These choices of keyword arguments produce a four panel plot with a single
# shared narrow colorbar on the right hand side of the multipanel plot. Axes
# labels are drawn for all plots since we're slicing along different directions
# for each plot.
grid = AxesGrid(fig, (0.075,0.075,0.85,0.85),
                nrows_ncols = (2, 1),
                axes_pad = 0.05,
                label_mode = "L",
                share_all = True,
                cbar_location="right",
                cbar_mode="single",
                cbar_size="3%",
                cbar_pad="0%")

for i, fn in enumerate(fns):
    # Load the data and create a single plot
    ds = yt.load(fn) # load data

    # Make a ProjectionPlot with a width of 34 comoving megaparsecs
    p = yt.ProjectionPlot(ds, 'z', 'density', width=(34, 'Mpccm'))

    # Ensure the colorbar limits match for all plots
    p.set_zlim('density', 1e-4, 1e-2)

    # This forces the ProjectionPlot to redraw itself on the AxesGrid axes.
    plot = p.plots['density']
    plot.figure = fig
    plot.axes = grid[i].axes
    plot.cax = grid.cbar_axes[i]

    # Finally, this actually redraws the plot.
    p._setup_plots()

plt.savefig('multiplot_1x2_time_series.png', bbox_inches='tight')

您也可以按照自己的方式进行操作（使用fig.add_subplots代替AxesGrid），但您需要手动定位轴并调整图形大小。

最后，如果您希望图形更小，则可以通过plt.figure()创建图形时以英寸为单位传递数字大小来控制图形的大小。如果您这样做，也可以通过调用p.set_font_size()上的ProjectionPlot来调整字体大小。