在python pandas中，如何保存'网格图'？

Question

我对pandas情节设施很新，在doc中，以下命令非常方便：

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')