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