python - 如何绘制图像及其类别的网格

Question

我正在尝试创建一个与此类似的情节。

我有几张图片，每张图片都有两个列表(两个列表都有 5 个元素)。

我想绘制图像，其右侧有两个列表。由于有几张图片需要执行此操作，因此如果可以以平铺/网格格式完成就好了。

我的数据如下:

class0[i] = ['example0', 'example1', .., 'example4']
class1[i] = ['example0', 'example1', .., 'example4']

Answer 1

这里有一个建议，使用几乎可以是任何形状的轴网格，以及一个用于类标签完美对齐的表格。

class0 = [['class0_{:d}'.format(i+1) for i in range(5)] for _ in range(10)]
class1 = [['class1_{:d}'.format(i+1) for i in range(5)] for _ in range(10)]

nrows, ncols = 5,2
fig, axs = plt.subplots(nrows, ncols*2, figsize=(2*ncols*2,2*nrows))
for img_ax,txt_ax,curr_class0,curr_class1 in zip(axs.flat[0::2], axs.flat[1::2], class0, class1):
    img_ax.imshow(np.random.random(size=(5,5)))

    text = [[c,d] for c,d in zip(curr_class0,curr_class1)]
    txt_ax.table(cellText=text, loc='center', edges='open')
    txt_ax.set_axis_off()