python - 在 x 的范围内运行 y 值的中值

Question

下面是我从两个 numpy 数组构建的散点图。

散点图示例

我想添加到此图中的是 y 在 x 范围内的运行中位数。我在一个例子中进行了 Photoshop:

修改后的散点图

具体来说，我需要两个值之间 x 轴上 1 个单位的 bin 中数据点的中值(这个范围在许多绘图之间会有所不同，但我可以手动调整它)。我很感激能为我指明正确方向的任何提示。

Answer 1

我会使用 np.digitize为您进行分类。这样您就可以轻松应用任何函数并设置您感兴趣的范围。

import numpy as np
import pylab as plt

N = 2000
total_bins = 10

# Sample data
X = np.random.random(size=N)*10
Y = X**2 + np.random.random(size=N)*X*10

bins = np.linspace(X.min(),X.max(), total_bins)
delta = bins[1]-bins[0]
idx  = np.digitize(X,bins)
running_median = [np.median(Y[idx==k]) for k in range(total_bins)]

plt.scatter(X,Y,color='k',alpha=.2,s=2)
plt.plot(bins-delta/2,running_median,'r--',lw=4,alpha=.8)
plt.axis('tight')
plt.show()

作为该方法多功能性的一个示例，让我们添加由每个 bin 的标准差给出的误差线:

running_std    = [Y[idx==k].std() for k in range(total_bins)]
plt.errorbar(bins-delta/2,running_median,
              running_std,fmt=None)