mnistからデータを取り出すところが間違っているのか、、

Question

前提・実現したいこと

mnistのmnist_784のデータで画像を表示させようとしたとき問題が生じました。

発生している問題・エラーメッセージ

 File "<stdin>", line 1, in <module>
  File "/home/＊＊＊/miniconda3/envs/ml/lib/python3.7/site-packages/pandas/core/frame.py", line 3458, in __getitem__
    indexer = self.columns.get_loc(key)
  File "/home/＊＊＊/miniconda3/envs/ml/lib/python3.7/site-packages/pandas/core/indexes/base.py", line 3361, in get_loc
    return self._engine.get_loc(casted_key)
  File "pandas/_libs/index.pyx", line 76, in pandas._libs.index.IndexEngine.get_loc
  File "pandas/_libs/index.pyx", line 82, in pandas._libs.index.IndexEngine.get_loc
TypeError: '(1, slice(None, None, None))' is an invalid key


#single_imageでpixel_valuesをスライスしようとしてみたところこのようなエラーが発生した。

該当のソースコード

data = datasets.fetch_openml(
    "mnist_784",
    version=1,
    return_X_y=True
)
pixel_values, targets = data

single_image = pixel_values[1,:].reshape(28,28)

試したこと

dataの中身を表示させたりしました。

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Answer 1

pixel_values がndarrayではなくDataFrameになっています。

single_image = pixel_values.values[1,:].reshape(28,28)

にしてください。

Comments

[ota12ken21]

できました！ありがとうございます