如何在一个绘图中绘制两组轮廓

Question

我正在绘制两组轮廓，如下所示：

plt.figure(figsize=(5,5))

plt.contourf(data_density[:,:,72], origin = 'upper', extent = (0,4.81,5,0), 
    levels=[0.6,0.7,0.8,0.9,1,1.1,1.2,1.3,1.4,1.5,1.6,1.7,1.8,1.9,2,2.1,2.2,2.3,2.4,2.5,2.6,2.7,2.8,2.9,3,3.1,3.2,3.3,3.4,3.5,3.6,3.7,3.8,3.9,4]) 

plt.colorbar()


plt.figure(figsize=(5,5))

plt.contourf(data_density[:,:,theta_index], origin = 'upper', extent = (0,4.81,5,0),
    levels=[0.6,0.7,0.8,0.9,1,1.1,1.2,1.3,1.4,1.5,1.6,1.7,1.8,1.9,2,2.1,2.2,2.3,2.4,2.5,2.6,2.7,2.8,2.9,3,3.1,3.2,3.3,3.4,3.5,3.6,3.7,3.8,3.9,4])

plt.colorbar()

我想将两个地块放在一个地块中（一个地块彼此相邻），但是我却做不到

谢谢您的帮助

Answer 1

您需要将图形分配给变量：

fig = plt.figure(figsize=(5,5))

然后创建一个网格：

from mpl_toolkits.axes_grid1 import AxesGrid
grid = AxesGrid(fig, 141, # similar to subplot(141)
                nrows_ncols = (1, 2),
                axes_pad = 0.05,
                label_mode = "1",
                )

left = grid[0]
right = grid[1]

# create a contourf on each grid element
left.contourf(data_density[:,:,72], origin = 'upper', extent ... ) 
right.contourf(data_density[:,:,theta_index], orig ...)

或者，您可以使用subplot 。