python - matplotlib 平均区间图

Question

我正在从 R 过渡到 python，并且希望绘制两个变量的平均线。它是将 x 变量分为 x 轴的区间和 y 轴的 y 变量的平均值的图。

例如，如果我有 1000 个点 (x1,y1) 到 (x1000, y1000)，并且想要绘制到 3 个箱中，我将有 3 个 x 间隔条，其中每个条都有 y 变量的平均值落入相应的区间。

有谁知道这个图叫什么，以及我如何在 python 中做到这一点？在 R 中，我使用“cut”命令，然后绘制剪切的 x,y。

谢谢!

Answer 1

对于后续问题，我们可以使用箱线图做一些更强大的事情。

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt

# simulate some artificial data
x = np.random.randn(1000,)
y = 5 * x ** 2 + np.random.randn(1000,)
data = pd.DataFrame(0.0, columns=['X', 'Y'], index=np.arange(1000))
data.X = x
data.Y = y

# now do your stuff
# ================================
# use the pandas 'cut' function
data['X_bins'] = pd.cut(data.X, 3)
data.set_index('X_bins', append=True, inplace=True)
data.drop('X', axis=1, inplace=True)
data.unstack(level=1).boxplot()