pandas read_csvでカテゴリ変数を読み込むにはどうすればよいですか？

Question

データフレームにデータをロードする際に次のエラーが発生し、カテゴリ変数に自動的に変換されます。

df = pd.read_csv(filepath_or_buffer=filename, 
       header=0, 
       index_col=False, 
       usecols=['col1', 'col2', 'col3'], 
       dtype={'col1': int, 
         'col2': 'category', 
         'col3': float}) 

TypeError: data type "category" not understood

Answer 1

現在、それが実装されていないが、あなたは次の回避策を使用することができます。

dtype={'col1': int, 
     'col2': pd.CategoricalDtype(['cat1', 'cat2', 'cat3']), 
     'col3': float} 

hereは一番簡単な方法は、それを変換され、まだオープン機能要求

Answer 2

です事実の後のカテゴリに、特にあなたがyoの後まで有効なカテゴリのすべてを知っていないかもしれないことを考えればあなたはファイルを読んでいます。

df = pd.read_csv(filepath_or_buffer=filename, usecols=['col1', 'col2', 'col3']) 
df['col3'] = df.col3.astype('category')