如何解决Matplotlib TypeError：只有整数标量数组可以转换为标量索引

Question

我正在尝试打印带有cifar-10数据标签的图形xlabel。然而，我不断得到这种类型的错误，如下所示。

TypeError: only integer scalar arrays can be converted to a scalar index

import tensorflow as tf 
from tensorflow import keras

from tensorflow.keras.datasets.cifar10 import load_data

import numpy as np
import matplotlib.pyplot as plt

tf.keras.datasets.cifar10.load_data()

(x_train,y_train),(x_test,y_test)=tf.keras.datasets.cifar10.load_data()

class_names = ['airplane', 'automobile', 'bird', 'cat', 'deer',
           'dog', 'frog', 'horse', 'ship', 'truck']

x_train=x_train/255.0
x_test=x_test/255.0

plt.figure(figsize=(10,10))
for i in range(25):
    plt.subplot(5,5,i+1)
    plt.xticks([])
    plt.yticks([])
    plt.grid(False)
    plt.imshow(x_train[i], cmap=plt.cm.binary)
    plt.xlabel(class_names[x_train[i]])
plt.show()

它说问题出在plt上。xlabel（class_name[x_train[i]]）这条线。请帮我解决这个问题。

Answer 1

这是简单的打字错误。您需要更改以下行。随着这一变化，一切都按预期进行。

从…起

plt.xlabel(class_names[x_train[i]])

到

plt.xlabel(class_names[y_train[i][0]])

完整的代码在这里供您参考。