如何在 Matplotlib 中为合并的双类别条形图精准添加垂直参考线

老丽吖_8666

老丽吖_8666

2026-03-25

745人浏览

原创

如何在 Matplotlib 中为合并的双类别条形图精准添加垂直参考线

本文详解如何在使用 seaborn.barplot 合并 call 与 put 的 cdf 条形图后,基于实际 strike 值准确定位并绘制 40% 和 80% 分位点的垂直参考线,避免因索引错位导致的线条偏移问题。

本文详解如何在使用 seaborn.barplot 合并 call 与 put 的 cdf 条形图后,基于实际 strike 值准确定位并绘制 40% 和 80% 分位点的垂直参考线,避免因索引错位导致的线条偏移问题。

在金融数据分析中,常需对比看涨(CALL)与看跌(PUT)期权的累积分布函数(CDF),并在同一图表中标注关键分位点(如 40%、80%),以直观识别流动性集中区域或行权价偏好。然而,直接将原始 DataFrame 拼接后调用 sns.barplot,会导致 x 轴为自动排序后的离散类别(由 STRIKE 唯一值构成),此时若仍用原始 DataFrame 的 .idxmin() 返回的原始索引(如 CALL.index[5])传入 plt.axvline(x=...),会因索引与绘图坐标系不匹配而造成垂直线严重错位——这是初学者最常见的陷阱。

根本原因在于:seaborn.barplot 在 x='STRIKE' 模式下,会将所有唯一 STRIKE 值去重、升序排列,并按此顺序分配整数位置(0, 1, 2, …)作为横坐标;而 CALL['Percentile'].sub(0.4).abs().idxmin() 返回的是原始 CALL DataFrame 中的行标签(label)或整数位置(iloc),二者语义完全不同。

✅ 正确做法是:先构建统一的 x 轴坐标映射表。以下为推荐实现流程:

1. 统一提取并排序所有 Strike 值

res = pd.concat([CALL.assign(hue='CALL'), PUT.assign(hue='PUT')], ignore_index=True)
all_strikes = sorted(res['STRIKE'].unique())  # 升序排列,确保 barplot 顺序一致

2. 精确查找各分位点对应 Strike 及其在统一坐标中的位置

# 分别计算 CALL/PUT 的目标 Strike 值(非索引!)
nearest_40_call = CALL.loc[(CALL['Percentile'] - 0.4).abs().idxmin(), 'STRIKE']
nearest_80_call = CALL.loc[(CALL['Percentile'] - 0.8).abs().idxmin(), 'STRIKE']
nearest_40_put  = PUT.loc[(PUT['Percentile'] - 0.4).abs().idxmin(), 'STRIKE']
nearest_80_put  = PUT.loc[(PUT['Percentile'] - 0.8).abs().idxmin(), 'STRIKE']

# 将 Strike 值映射为 barplot 的横坐标位置(0-based 整数索引)
def strike_to_position(strike_val, strike_list):
    try:
        return strike_list.index(strike_val)  # 精确匹配
    except ValueError:
        # 若存在浮点精度误差,可改用近似匹配(如 round(strike_val, 2))
        idx = np.argmin(np.abs(np.array(strike_list) - strike_val))
        return idx

x_40c = strike_to_position(nearest_40_call, all_strikes)
x_80c = strike_to_position(nearest_80_call, all_strikes)
x_40p = strike_to_position(nearest_40_put,  all_strikes)
x_80p = strike_to_position(nearest_80_put,  all_strikes)

3. 绘制条形图 + 垂直线 + 标注文本

plt.figure(figsize=(16, 6))
ax = sns.barplot(data=res, x='STRIKE', y='CDF', hue='hue', 
                 palette={'CALL': 'lightgreen', 'PUT': 'orange'}, 
                 order=all_strikes)  # 显式指定 order,确保坐标可预测

# 添加垂直线(x 为整数位置)
plt.axvline(x=x_40c, color='red', linestyle='--', linewidth=1.2, alpha=0.8)
plt.axvline(x=x_80c, color='red', linestyle='--', linewidth=1.2, alpha=0.8)
plt.axvline(x=x_40p, color='blue', linestyle='--', linewidth=1.2, alpha=0.8)
plt.axvline(x=x_80p, color='blue', linestyle='--', linewidth=1.2, alpha=0.8)

# 添加带颜色的文本标注(置于顶部,避免遮挡)
y_max = res['CDF'].max() * 1.05
plt.text(x_40c, y_max, f'40% CALL\n{nearest_40_call}', 
         ha='center', va='bottom', color='red', fontsize=9, fontweight='bold')
plt.text(x_80c, y_max, f'80% CALL\n{nearest_80_call}', 
         ha='center', va='bottom', color='red', fontsize=9, fontweight='bold')
plt.text(x_40p, y_max, f'40% PUT\n{nearest_40_put}', 
         ha='center', va='bottom', color='blue', fontsize=9, fontweight='bold')
plt.text(x_80p, y_max, f'80% PUT\n{nearest_80_put}', 
         ha='center', va='bottom', color='blue', fontsize=9, fontweight='bold')

# 自定义图例(去除 seaborn 自动生成的 hue 图例,避免冗余)
handles = [plt.Rectangle((0,0),1,1, color='lightgreen', ec="k"),
           plt.Rectangle((0,0),1,1, color='orange', ec="k")]
plt.legend(handles, ['CALL', 'PUT'], title='Option Type', loc='upper right')

plt.xlabel('Strike Price', fontsize=11)
plt.ylabel('Cumulative Distribution Function (CDF)', fontsize=11)
plt.title('CDF Comparison: CALL vs PUT with 40% & 80% Percentile Lines', fontsize=12, pad=20)
plt.xticks(rotation=45, ha='right')
plt.grid(True, axis='y', alpha=0.3)
plt.tight_layout()
plt.show()

⚠️ 关键注意事项:

  • 勿混用索引与坐标:axvline(x=...) 中的 x 必须是绘图坐标系中的数值(对 barplot(x='col') 即 0, 1, 2,...),而非原始 DataFrame 行号。
  • 显式指定 order= 参数:sns.barplot(..., order=all_strikes) 确保条形顺序与 all_strikes 严格一致,是定位可靠的前提。
  • 处理浮点 Strike:若 STRIKE 为浮点数(如 2850.5),.index() 可能因精度失败,应改用 np.argmin(np.abs(...)) 近似匹配。
  • 标注位置优化:使用 plt.text() 并设置 ha='center' 和动态 y_max,比固定 va='top' 更鲁棒;颜色与线条保持一致提升可读性。
  • 性能提示:对于超大 STRIKE 列表,可用 bisect 模块加速查找,但通常 list.index() 已足够高效。

通过建立“Strike 值 → 绘图位置”的明确映射,即可在多类别合并图表中实现任意分位线的像素级精准标注,兼顾专业性与可复现性。

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

2023.07.20

1831

4

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

2023.07.25

4904

7

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.07.31

1789

3

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

2023.08.03

29317

23

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

3507

5

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

3507

5

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

1223

5

python合并两个列表
python合并两个列表

Python是一种强大的编程语言,具有许多方便的功能和工具。在Python中,有多种方法可以合并两个列表。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.10

616

4

python是前端还是后端
python是前端还是后端

Python属于前端也属于后端,其灵活性和丰富的生态系统使得开发人员能够在不同的领域中灵活运用。本专题为大家提供python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

2723

5

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
热门推荐
/
最新课程
phpStudy极速入门视频教程
phpStudy极速入门视频教程

共6课时 | 54.7万人学习

独孤九贱(4)_PHP视频教程
独孤九贱(4)_PHP视频教程

共89课时 | 134.1万人学习