KeyError同时获取MNIST数据

Question

我正在尝试处理MNSIT数据集，这是我在一开始就遇到的错误。获取MNIST图像时出错

from sklearn.datasets import fetch_openml
mnist = fetch_openml('mnist_784')
x, y = mnist['data'], mnist['target']
digit = x[36001]

Answer 1

x是二维的，请尝试如下：

digit = x[36001, :]

因为，您正面临一个错误，这肯定是由于x没有被视为numpy array，所以请尝试以下两种方法：

digit = x.iloc[36001, :]

或

digit = x.to_numpy()[36001,:]

Answer 2

扩大我的评论，我认为MNIST数据集的openml最近是(?)切换到返回熊猫DataFrame，而不是numpy数组。另一个可能的修复是在函数调用中：

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