python - 如何在Python中绘制小平面网格(R中的ggplot)

Question

我希望使用 python(matplotlib) 绘制以下图表。

我可以通过以下代码在 R 中执行此操作，

ggplot(delay.with.weather,
aes(x=pressure, y=dep.delay), na.rm = TRUE) + geom_point(na.rm = TRUE) +
facet_grid(season ~ ., scales="free") + stat_smooth(method=lm, na.rm = TRUE)

我的尝试如下，

plt.scatter(flights_combined.pressure,flights_combined.dep_delay)
plt.show()

我无法按季节列分组。也无法绘制回归线。有人可以帮我做这些吗？

Answer 1

我建议使用seaborn在 matplotlib 之上进行此类可视化，因为它与 Dataframe 配合得很好。

您可以通过以下方式实现分组散点图:

grid = seaborn.FacetGrid(flights_combined, row='season')
grid.map(plt.scatter, 'pressure', 'delay')

请参阅 FacetGrid 的文档了解更多详情。

如果您想要回归线，您可以手动执行(创建您自己的 plt.scatter 实现来拟合并在数据上绘制回归线)，或者您可以找到一种使用 lmplot 的方法。 ( Example ) 与 FacetGrid 结合.