python - 在 matplotlib 中绘制多个图时出现索引错误

Question

我正在尝试通过子图绘制多个图。代码“有效”，但它总是给我一个索引错误，我一生都无法弄清楚。

作为一个附带问题，我想知道是否有人知道如何使每个单独的地块保持相同的大小。例如，如果我添加更多行或列，那么每个图都会变小。谢谢。

count = 0
n_rows = 2
n_columns = 2
f, axarr = plt.subplots(n_rows, n_columns)
plt.figure(figsize=(20,20))

for column in range(n_cols):
    for row in range(n_rows):
        axarr[row, column].imshow(generate_pattern('block3_conv1', count, size=150))

        count += 1

错误

IndexError                                Traceback (most recent call last)
<ipython-input-37-7f7ae19e07e9> in <module>()
      7 for column in range(n_cols):
      8     for row in range(n_rows):
----> 9         axarr[row, column].imshow(generate_pattern('block3_conv1', count, size=150))
     10 
     11         count += 1

IndexError: index 2 is out of bounds for axis 1 with size 2

使用的函数代码

def generate_pattern(layer_name, filter_index, size=150):
    # Build a loss function that maximizes the activation
    # of the nth filter of the layer considered.
    layer_output = model.get_layer(layer_name).output
    loss = K.mean(layer_output[:, :, :, filter_index])

    # Compute the gradient of the input picture wrt this loss
    grads = K.gradients(loss, model.input)[0]

    # Normalization trick: we normalize the gradient
    grads /= (K.sqrt(K.mean(K.square(grads))) + 1e-5)

    # This function returns the loss and grads given the input picture
    iterate = K.function([model.input], [loss, grads])

    # We start from a gray image with some noise
    input_img_data = np.random.random((1, size, size, 3)) * 20 + 128.

    # Run gradient ascent for 40 steps
    step = 1.
    for i in range(40):
        loss_value, grads_value = iterate([input_img_data])
        input_img_data += grads_value * step

    img = input_img_data[0]
    return deprocess_image(img)

def deprocess_image(x):
    x -= x.mean()
    x /= (x.std() + 1e-5)
    x *= 0.1

    x += 0.5
    x = np.clip(x,0,1)

    x *= 255
    x = np.clip(x,0,255).astype('uint8')

    return x  

标签： pythonmatplotlib

Answer 1

此错误是由于尝试plt.subplots()使用超出索引范围的值来索引生成的数组而导致的。显示这一点的一种方法是用数字替换循环中的变量。在这种情况下，您将看到axarr[1,2]将产生以下错误：

---------------------------------------------------------------------------
IndexError                                Traceback (most recent call last)
<ipython-input-39-31f90736bd1d> in <module>()
      2 #plt.figure(figsize=(20,20))
      3 
----> 4 a[0,2]

IndexError: index 2 is out of bounds for axis 1 with size 2

我们知道函数中没有发生错误，generate_pattern因为错误消息会表明很多。