Матплотлиб гист метод

Question

Я пытаюсь построить гистограмму набора данных MNIST:

numberMNIST = fetch_openml('mnist_784',return_X_y=False)

dataset = numberMNIST.data     
labels = numberMNIST.target    

X_train, X_test, Y_train, Y_test = dataset[:60000], dataset[60000:], labels[:60000], labels[60000:]
Y_train_is4 = (Y_train == '4')
Y_test_is4 = (Y_test == '4')

plt.hist(Y_train)
plt.xlabel("Label")
plt.ylabel("Quantity")
plt.title("Labels in MNIST 784 dataset")

Однако порядок меток организован неправильно:

Как я могу решить это, увидев последовательность 0,1,2,3, ..., 9 по оси X?

Answer 1

hist функция больше подходит для числовых c переменных. В вашем случае метки являются строками и рассматриваются как object по numpy. Я бы предпочел использовать бар .

import numpy as np

unique, counts = np.unique(Y_train, return_counts=True)
plt.bar(unique, counts)
plt.xticks(unique)
plt.xlabel("Label")
plt.ylabel("Quantity")
plt.title("Labels in MNIST 784 dataset")