python - 将 matplotlib 图保存为数据库中的图像字段

Question

我刚开始学习matplotlib，我想在我的一个 django 应用程序中使用它。所以我想知道如何保存在我的模型的图像字段中生成的图形，以便我可以在需要时进行检索。

Answer 1

matplotlib.pyplot.savefig接受类文件对象作为第一个参数。您可以传递StringIO/ BytesIO（根据您的 python 版本）。

f = StringIO()
plt.savefig(f)

然后，使用django.core.files.ContentFile将字符串转换为django.core.files.File（因为FieldFile.save只接受一个实例django.core.files.File）。

content_file = ContentFile(f.getvalue())
model_object = Model(....)
model_object.image_field.save('name_of_image', content_file)
model_object.save()