基于样本量调整线宽的x-平均y关系绘图及Matplotlib实现咨询

这类图的名称 #

这类图通常被称为样本量加权均值折线图，也可宽泛归为加权折线图的一种——核心是用线条粗细反映x对应y均值的样本量大小，以此体现均值的统计可靠性：样本量越大，线条越粗，代表均值可信度越高；反之则越细，提示该均值参考价值较低。

Matplotlib实现方法 #

完全可以用Matplotlib实现，核心思路是先按x分组计算均值和样本量，再根据样本量动态设置折线粗细。以下是具体步骤和示例代码：

步骤说明 #

1. 按x值分组，计算每个x对应的y平均值及每组样本量（行数）。
2. 将分组后的数据按x排序，保证折线绘制顺序正确。
3. 对样本量做归一化处理（避免样本量差距过大导致线条粗细极端），将其作为折线的linewidth参数值。
4. 可选：添加样本量标注或图例，让读者直观理解线条粗细的含义。

示例代码 #

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

# 生成模拟数据：x取5个值，每个x对应不同数量的y值
np.random.seed(42)
x_values = np.repeat([1, 2, 3, 4, 5], [20, 5, 30, 8, 15])
y_values = x_values * 2 + np.random.normal(0, 1, size=len(x_values))
df = pd.DataFrame({'x': x_values, 'y': y_values})

# 按x分组计算均值和样本量
grouped = df.groupby('x').agg(
    y_mean=('y', 'mean'),
    sample_size=('y', 'count')
).reset_index()

# 对样本量做归一化，控制线条粗细在1-5之间
max_sample = grouped['sample_size'].max()
grouped['line_width'] = (grouped['sample_size'] / max_sample) * 4 + 1

# 绘制折线图
plt.figure(figsize=(10, 6))
# 逐段绘制折线，实现每段线条粗细随样本量变化
for i in range(len(grouped)-1):
    x_segment = [grouped['x'].iloc[i], grouped['x'].iloc[i+1]]
    y_segment = [grouped['y_mean'].iloc[i], grouped['y_mean'].iloc[i+1]]
    plt.plot(x_segment, y_segment, color='blue', linewidth=grouped['line_width'].iloc[i])

# 绘制每个x对应的均值点
plt.scatter(grouped['x'], grouped['y_mean'], color='red', zorder=5)

# 添加样本量标注
for idx, row in grouped.iterrows():
    plt.text(row['x'] + 0.1, row['y_mean'], f'n={row["sample_size"]}', fontsize=9)

# 设置图表标签
plt.xlabel('x')
plt.ylabel('Mean of y')
plt.title('Mean of y vs x (Line Width = Sample Size)')

# 添加粗细说明的图例
from matplotlib.lines import Line2D
custom_lines = [
    Line2D([0], [0], color='blue', lw=1),
    Line2D([0], [0], color='blue', lw=5)
]
plt.legend(custom_lines, [f'Small sample (min={grouped["sample_size"].min()})', f'Large sample (max={max_sample})'])

plt.show()

代码说明 #

• 用groupby快速完成分组统计，得到每个x对应的y均值和样本量。
• 归一化样本量是为了避免样本量差距过大时，线条粗细超出视觉舒适范围（示例中将粗细控制在1到5之间）。
• 逐段绘制折线是因为Matplotlib默认对整条折线用统一粗细，逐段绘制可实现每段线条粗细随样本量变化，可根据需求调整对应逻辑（比如用终点样本量控制线段粗细）。

内容的提问来源于stack exchange，提问作者gillenpj