2018年9月14日金曜日

Pandasでcsvをread_csvを使って読み込むときに列名にスペースが入るので注意

Pandasでcsvをread_csvを使って読み込むとき、何も設定しなければ1行めがheaderに設定できますが、注意が必要です、という話。
test.csvname, num
AAA, 10
BBB, 20
CCC, 30
だったとして、
import pandas as pd
df = pd.read_csv('test.csv')
とやってデータフレームを読み込みます
# In
df.columns
# Out
Index(['name', ' num'], dtype='object')
・・・何かおかしいのですが、カラム名のnum の前にスペースが入っています
ということで、
df['num']
とやると
KeyError: 'num'
が出てしまいますので注意。csv側で1行めはスペースを抜くか、names=['name', 'num']で指定するのがよいようです。

0 件のコメント:

コメントを投稿