下面是我从两个 numpy 数组构建的散点图。
散点图示例
我想添加到该图中的是 y 在 x 范围内的运行中位数。我已经在一个示例中进行了 Photoshop 处理：
修改散点图
具体来说，我需要两个值之间沿 x 轴 1 个单位的箱中的数据点的中位数（这个范围在许多图之间会有所不同，但我可以手动调整它）。我感谢任何可以为我指明正确方向的提示。
【问题讨论】：
标签： python numpy matplotlib median scatter
我会使用 np.digitize 为您进行 bin 排序。这样您就可以轻松应用任何功能并设置您感兴趣的范围。
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)
【讨论】：
这个问题也可以通过python pandas（Python数据分析库）有效解决，提供原生的数据切割和分析方法。
考虑一下
（感谢 @Hooked 并为他的示例 +1，我从中借用了 X 和 Y 数据）
import pandas as pd
df = pd.DataFrame({'X' : X, 'Y' : Y}) #we build a dataframe from the data
data_cut = pd.cut(df.X,bins) #we cut the data following the bins
grp = df.groupby(by = data_cut) #we group the data by the cut
ret = grp.aggregate(np.median) #we produce an aggregate representation (median) of each bin
#plotting
plt.scatter(df.X,df.Y,color='k',alpha=.2,s=2)
plt.plot(ret.X,ret.Y,'r--',lw=4,alpha=.8)
plt.show()
备注：这里红色曲线的x值是bin-wise x-median（可以使用bin的中点）。
【讨论】：
您可以基于numpy.median() 创建一个函数，该函数将计算给定区间的中值：
import numpy as np
def medians(x, y, intervals):
out = []
for xmin, xmax in intervals:
mask = (x >= xmin) & (x < xmax)
out.append(np.median(y[mask]))
return np.array(out)
然后使用此函数获得所需的间隔：
import matplotlib.pyplot as plt
intervals = ((18, 19), (19, 20), (20, 21), (21, 22))
centers = [(xmin+xmax)/2. for xmin, xmax in intervals]
plt.plot(centers, medians(x, y, intervals)
【讨论】：
intervals = pairwise((18, 19, 20, 21, 22)) 或 pairwise(range(18, 23, 1))
我在C# 中写过类似的内容。我不做 Python 所以这里是伪代码：
List 用于从中派生中位数的数据x 值对散点图点进行排序x值循环排序点Y 值插入到中值列表中，以便中值列表作为排序列表增长。即插入 Y，使其上方和下方的 List 值分别为 > 和 Inserting values into specific locations in a list in Python。Y 值添加后，中间值将是当前中间索引处的列表值，即List(List.Length/2)
希望对你有帮助！
【讨论】：