Напишите CSV с dtype

Question

Я хочу сохранить большой CSV с помощью панд, однако, когда я сохраняю его, я теряю информацию dtype, и, следовательно, при чтении я получаю ошибку: DtypeWarning: Columns ... have mixed types. Specify dtype option on import or set low_memory=False.

Это какой-нибудь элегантный способ писать и читать csv и сохранять dtypes?

Answer 1

Вы можете попробовать импортировать набор данных следующим образом:

dtypes = {
    'col_name1': 'type',
    'col_name2': 'type',
...
}
df = pd.read_csv('df_name.csv', sep = ';', encoding = 'cp1251', dtype = dtypes)

использовать sep и кодировку набора данных

при сохранении, используйте:

df.to_csv('df_name.csv', index = False)