python - 在 python pandas 中，如何保存 'grid chart' ？

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