python - 在 python 中绘制配置文件命中图

Question

我正在尝试为 pandas.DataFrame 的两列绘制剖面图。我不希望它直接出现在 pandas 中，但似乎 matplotlib 中也没有。我四处搜索，但在 rootpy 以外的任何包中都找不到它。在我花时间自己写这篇文章之前，我想我会问是否有一个包含配置文件直方图的小包，也许它们以不同的名称为人所知。

如果您不知道我所说的“配置文件直方图”是什么意思，请查看 ROOT 实现。 http://root.cern.ch/root/html/TProfile.html

Answer 1

您可以使用 scipy.stats.binned_statistic 轻松完成此操作.

import scipy.stats
import numpy
import matplotlib.pyplot as plt

x = numpy.random.rand(10000)
y = x + scipy.stats.norm(0, 0.2).rvs(10000)

means_result = scipy.stats.binned_statistic(x, [y, y**2], bins=50, range=(0,1), statistic='mean')
means, means2 = means_result.statistic
standard_deviations = numpy.sqrt(means2 - means**2)
bin_edges = means_result.bin_edges
bin_centers = (bin_edges[:-1] + bin_edges[1:])/2.

plt.errorbar(x=bin_centers, y=means, yerr=standard_deviations, linestyle='none', marker='.')