jupyter の plt.imshow()で「KeyError: 0」が出てしまう

Question

[作りながら学ぶ！深層強化学習｛PyTorchによる実践プログラミング｝]
にあるコードを入力しています。
MNISTダウンロードする必要があり

#手書き数字の画像データMNISTをダウンロード
from sklearn.datasets import fetch_openml
mnist = fetch_openml('mnist_784', version=1)

#1. データの前処理（画像データとラベルに分割し。正規化）
X = mnist.data / 255 #0-255を0-1に正規化
y = mnist.target

#MNISTのデータの一つ目を可視化する
import matplotlib.pyplot as plt
%matplotlib inline

plt.imshow(X[0].reshape(28, 28), cmap='gray')
print("この画像データのラベルは{:.0f}です".format(y[0]))

と入力したところ

KeyError                                  Traceback (most recent call last)
File ~\anaconda3\envs\rl_env\lib\site-packages\pandas\core\indexes\base.py:3621, in Index.get_loc(self, key, method, tolerance)
   3620 try:
-> 3621     return self._engine.get_loc(casted_key)
   3622 except KeyError as err:

File ~\anaconda3\envs\rl_env\lib\site-packages\pandas\_libs\index.pyx:136, in pandas._libs.index.IndexEngine.get_loc()

File ~\anaconda3\envs\rl_env\lib\site-packages\pandas\_libs\index.pyx:163, in pandas._libs.index.IndexEngine.get_loc()

File pandas\_libs\hashtable_class_helper.pxi:5198, in pandas._libs.hashtable.PyObjectHashTable.get_item()

File pandas\_libs\hashtable_class_helper.pxi:5206, in pandas._libs.hashtable.PyObjectHashTable.get_item()

KeyError: 0

The above exception was the direct cause of the following exception:

KeyError                                  Traceback (most recent call last)
Input In [3], in <cell line: 5>()
      2 import matplotlib.pyplot as plt
      3 get_ipython().run_line_magic('matplotlib', 'inline')
----> 5 plt.imshow(X[0].reshape(28, 28), cmap='gray')
      6 print("この画像データのラベルは{:.0f}です".format(y[0]))

File ~\anaconda3\envs\rl_env\lib\site-packages\pandas\core\frame.py:3505, in DataFrame.__getitem__(self, key)
   3503 if self.columns.nlevels > 1:
   3504     return self._getitem_multilevel(key)
-> 3505 indexer = self.columns.get_loc(key)
   3506 if is_integer(indexer):
   3507     indexer = [indexer]

File ~\anaconda3\envs\rl_env\lib\site-packages\pandas\core\indexes\base.py:3623, in Index.get_loc(self, key, method, tolerance)
   3621     return self._engine.get_loc(casted_key)
   3622 except KeyError as err:
-> 3623     raise KeyError(key) from err
   3624 except TypeError:
   3625     # If we have a listlike key, _check_indexing_error will raise
   3626     #  InvalidIndexError. Otherwise we fall through and re-raise
   3627     #  the TypeError.
   3628     self._check_indexing_error(key)

KeyError: 0

と出ました

簡単に言えば、MNISTのダウンロードがうまくいってないってことでしょうか。

Answer 1

fetch_openmlで返されるデータ型がnumpyからpandasに変わっているようです。
よって以下のようにnumpy型に変換するとよいでしょう。

略
mnist = fetch_openml('mnist_784', version=1)
mnist.data = mnist.data.values
mnist.target = mnist.target.astype(float).values
略

Comments

[Tyutohannpa_]

解決できました。本当にありがとうございます。

なぜ返されるデータ型がnumpyからpandasに代わってしまったのでしょうか。。差支えなければ教えていただきたいです。。。

[can110]

理由については分かりません。そういった要望がsklearnに寄せられたのかもしれません。

[melian]

> 返されるデータ型がnumpyからpandasに代わってしまったのでしょうか
API ドキュメントを読むと、as_frame=False を指定すると numpy.ndarray になるそうです。

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

as_frame のデフォルト値は "auto" で、データによって適切に？決まる様です。