在Python pandas中，如何保存“网格图”？

Question

我对pandas的绘图功能还很陌生，在文档中，下面这个命令非常实用：

myplot = rts.ret.hist(bins=50, by=rts.primary_mic)

但是，当我尝试从绘图中获取图形引用并保存时，就遇到了问题：

myfigure = myplot.get_figure()
AttributeError: 'numpy.ndarray' object has no attribute 'get_figure'

我理解的是，rts.ret.hist(bins=50) 返回的是一个图形对象，而rts.ret.hist(bins=50 返回的是一个数组对象。

在这种情况下，我应该怎么保存我的图形呢？

有没有什么线索？

谢谢！

Answer 1

要保存这个图形，你可以使用 plt.savefig 这个命令：

import pandas as pd
import matplotlib.pyplot as plt

df = pd.DataFrame(
    [(1, 2), (1, 3), (1, 4), (2, 1), (2, 2)], columns=['col1', 'col2'])
df.hist(bins=4, by=df['col1'])
plt.savefig('/tmp/out.png')