如何在mplfinance散点图中绘制多类型标记并解决addplot报错

问题原因 #

直接将df['Flag']传入addplot参数会触发TypeError，因为mplfinance的addplot不接受原始整数Series，它要求传入由mpf.make_addplot()构造的绘图对象，或符合规范的对象列表/字典。

实现步骤 #

以下是基于pandas 1.5.3 + Python 3.10的可行方案，包含两种实现方式：

1. 依赖导入 #

import pandas as pd
import mplfinance as mpf
import matplotlib.pyplot as plt

2. 数据准备（替换为你的真实数据） #

# 构造示例时间序列数据（datetime索引+OHLC+Flag列）
dates = pd.date_range(start='2023-01-01', periods=20, freq='D')
df = pd.DataFrame({
    'Open': range(100, 120),
    'High': range(105, 125),
    'Low': range(95, 115),
    'Close': range(100, 120),
    'Flag': [0,1,0,2,0,3,0,1,0,2,0,3,0,1,0,2,0,3,0,0]
}, index=dates)

3. 定义Flag与标记的映射 #

# 整数Flag对应matplotlib标记
flag_to_marker = {
    1: '^',  # 向上三角
    2: 'v',  # 向下三角
    3: 'o'   # 圆形
}

# 可选：为不同标记定义颜色
marker_colors = {
    1: 'green',
    2: 'red',
    3: 'orange'
}

方法一：使用回调函数绘制自定义标记（推荐） #

通过post_plot_callback在蜡烛图绘制完成后，直接在轴上添加散点标记，无需拆分数据：

def plot_flag_markers(ax, data, marker_map, color_map):
    for idx, row in data.iterrows():
        flag = row['Flag']
        if flag in marker_map:
            # 计算当前日期在x轴的位置（基于索引偏移）
            x_pos = ax.get_xlim()[0] + (idx - data.index[0]).days
            # 用收盘价作为y轴位置，可替换为High/Low等
            y_pos = row['Close']
            # 绘制标记，zorder确保标记在蜡烛图上方
            ax.scatter(x_pos, y_pos, marker=marker_map[flag], 
                       color=color_map.get(flag, 'black'), s=100, zorder=5)

# 绘制蜡烛图并绑定回调
mpf.plot(df, type='candle', style='yahoo',
         title='Candlestick with Flag Markers',
         post_plot_callback=lambda ax: plot_flag_markers(ax, df, flag_to_marker, marker_colors))
plt.show()

方法二：拆分数据用原生addplot添加 #

将每个Flag对应的位置拆分为独立Series，通过mpf.make_addplot构造绘图对象后传入：

addplot_list = []
for flag, marker in flag_to_marker.items():
    # 仅保留当前Flag对应的价格值，其余设为NaN
    marker_series = df['Close'].where(df['Flag'] == flag)
    # 创建散点类型的addplot
    ap = mpf.make_addplot(marker_series, type='scatter', 
                          marker=marker, color=marker_colors[flag], markersize=100)
    addplot_list.append(ap)

# 绘制蜡烛图并加载所有addplot
mpf.plot(df, type='candle', style='yahoo',
         title='Candlestick with Flag Markers',
         addplot=addplot_list)
plt.show()

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