plt.scatterでc=yとすると何故かエラーがでる

Question

MNISTデータセットでグラフに色付けしたいと思ったのですが、
なぜかエラーが出ます。理由を教えてほしいです。

import numpy as np
import matplotlib.pyplot as plt
from sklearn.manifold import TSNE
from sklearn.datasets import fetch_openml


mnist = fetch_openml('mnist_784', version=1, as_frame=False)

#最初の6万個のデータからランダムで１万個抽出
np.random.seed(42)

m=10000
index = np.random.permutation(60000)[:m]

X = mnist.data[index]
y = mnist.target[index]

tsne = TSNE(n_components=2, random_state=42, n_jobs=-1)
X_reduced = tsne.fit_transform(X)

plt.scatter(X_reduced[:, 0], X_reduced[:, 1], c=y, cmap='viridis')
plt.axis('off')
plt.colorbar()
plt.show()

ValueError: 'c' argument must be a color, a sequence of colors, or a sequence of numbers, not ['7' '3' '8' ... '9' '8' '1']

Answer 1

ValueError: 'c' argument must be a color, a sequence of colors, or a sequence of numbers, not ['7' '3' '8' ... '9' '8' '1']

y が文字列のリストになっていますので int のリストに変換します。

#y = mnist.target[index]
y = list(map(int, mnist.target[index]))