用dtype编写csv

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)