如何在文件字段中将数据框保存为 csv 文件

Question

我正在尝试将数据帧保存为模型对象的文件字段中的 csv 文件，但保存不正确，正在保存的文件包含一些其他语言字符！！请告诉我做错了什么？？

new_df = df.to_csv(columns=['A', 'B'], index=False)
doc.csvfile.save(f'{doc.id}.csv', ContentFile(new_df))

Answer 1

您好，您可以尝试使用以下代码保存csv文件

import csv
from io import StringIO
from django.core.files.base import ContentFile

new_df = df.to_csv(columns=['A', 'B'], index=False)

csv_buffer = StringIO()
csv_writer = csv.writer(csv_buffer)
csv_writer.writerow(new_df)

csv_file = ContentFile(csv_buffer.getvalue().encode('utf-8'))
doc.csvfile.save('output.csv', csv_file)

Answer 2

您可以简单地使用 pandas 数据帧的 to_csv 方法在内存中创建一个临时文件。

这里不需要使用任何字符串缓冲机制。

from django.core.files.base import ContentFile

content = df.to_csv(columns=['A', 'B'], index=False)
temp_file = ContentFile(content.encode('utf-8'))

doc.csvfile.save(f'{doc.id}.csv', temp_file)

如何在文件字段中将数据框保存为 csv 文件

How to save a dataframe as csv file in filefield

django

pandas

dataframe

django-models

filefield