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并自动引用分类数据?