Pandas個人的注意点まとめ

概要

自分用に使っていて注意したほうがいい点をまとめる。
順次追加予定。

indexの型に関して

CSVの読み込み時index_colを利用するとindexの型はdtypeの指定とは別に
自動で決定されてしまうようだ。

import pandas as pd

# 例) input.csv
# ID,param1,param2
# 0001,01,AAA
# 0002,02,BBB
# 0003,10,CCC

df = pd.read_csv("input.csv", dtype=object, index_col="ID")
# 型をobjectに指定しているがindexだけはint型となってしまい
# 先頭の0が抜け落ちてしまう

型を保持するためにはindexの指定を後から行う必要があるようだ。

df = pd.read_csv("input.csv", dtype=object)
df.set_index("ID", inplace=True)
# これで元のデータのまま保持できる