熊猫:在导出为CSV时如何引用分类数据

时间:2018-09-22 21:22:48

标签: pandas export-to-csv

Pandas数据框to_csv方法具有引用非数字数据的选项。不幸的是,它没有引用分类列。

这是正在发生的事情:

In [32]: df = pd.DataFrame({'a': list('abc'), 
                 'b': range(3), 'c': range(3)})

In [33]: df['c'] = df['c'].astype('category')

In [34]: df.dtypes
Out[34]: 
a      object
b       int64
c    category
dtype: object

In [35]: import csv

In [36]: df.to_csv("test.csv", quoting=csv.QUOTE_NONNUMERIC)
In [39]: !head test.csv
"","a","b","c"
0,"a",0,0
1,"b",1,1
2,"c",2,2

我希望c列周围有引号。

如何将数据框导出到csv并自动引用分类数据?

0 个答案:

没有答案