Панды: как процитировать категориальные данные при экспорте в CSV - PullRequest
Новая
       8

Панды: как процитировать категориальные данные при экспорте в CSV

0 голосов
/ 23 сентября 2018

Pandas dataframe 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 и автоматически указывать мои категориальные данные?

...