問題
Pandas で CSV を読み込みたくて次のコードを実行したところ…
import pandas as pd
df = pd.read_csv('bmi.csv')
label = df['label']
print(label)
以下のエラーが出ました。
> Traceback (most recent call last):
>
> File "C:\Users\user\Anaconda3\lib\site-packages\pandas\core\indexes\base.py", line 3063, in get_loc
>
> return self.\_engine.get_loc(key)
>
> File "pandas\\\_libs\index.pyx", line 140, in pandas.\_libs.index.IndexEngine.get_loc
>
> File "pandas\\\_libs\index.pyx", line 162, in pandas.\_libs.index.IndexEngine.get_loc
>
> File "pandas\\\_libs\hashtable_class_helper.pxi", line 1492, in pandas.\_libs.hashtable.PyObjectHashTable.get_item
>
> File "pandas\\\_libs\hashtable_class_helper.pxi", line 1500, in pandas.\_libs.hashtable.PyObjectHashTable.get_item
>
> KeyError: 'label'
>
> During handling of the above exception, another exception occurred:
>
> Traceback (most recent call last):
>
> File "C:/Users/user/PycharmProjects/kujirabon/practice/bmi-test.py", line 10, in
>
> label = tbl['label']
>
> File "C:\Users\user\Anaconda3\lib\site-packages\pandas\core\frame.py", line 2685, in \_\_getitem\_\_
>
> return self.\_getitem_column(key)
>
> File "C:\Users\user\Anaconda3\lib\site-packages\pandas\core\frame.py", line 2692, in \_getitem_column
>
> return self.\_get_item_cache(key)
>
> File "C:\Users\user\Anaconda3\lib\site-packages\pandas\core\generic.py", line 2486, in \_get_item_cache
>
> values = self.\_data.get(item)
>
> File "C:\Users\user\Anaconda3\lib\site-packages\pandas\core\internals.py", line 4115, in get
>
> loc = self.items.get_loc(item)
>
> File "C:\Users\user\Anaconda3\lib\site-packages\pandas\core\indexes\base.py", line 3065, in get_loc
>
> return self.\_engine.get_loc(self.\_maybe_cast_indexer(key))
>
> File "pandas\\\_libs\index.pyx", line 140, in pandas.\_libs.index.IndexEngine.get_loc
>
> File "pandas\\\_libs\index.pyx", line 162, in pandas.\_libs.index.IndexEngine.get_loc
>
> File "pandas\\\_libs\hashtable_class_helper.pxi", line 1492, in pandas.\_libs.hashtable.PyObjectHashTable.get_item
>
> File "pandas\\\_libs\hashtable_class_helper.pxi", line 1500, in pandas.\_libs.hashtable.PyObjectHashTable.get_item
>
> KeyError: 'label'
bmi.csvの中身
| height | weight | label |
|---|---|---|
| 180 | 80 | normal |
| 172 | 65 | normal |
解決方法
どうしたものかと思って色々ググっていたところ、同じ問題が StackOverflow にありました。
python - Problems while trying to read a csv with pandas? - Stack Overflow
これによると CSV のヘッダーが、weight, height, label となっているとき、label を指定したければ、,の後ろの半角スペースまで入れる必要があるということでした。
つまり、label を指定したければ、df['label']と記述する必要があるということです。
また、別の解法として、read_csvするときにオプションでskipinitialspace=Trueを追加してあげれば df['label']としても問題ありませんでした。