mplfinance绘制K线叠加买卖信号报错：x与y维度不匹配

核心原因 #

分开绘制买卖信号时触发ValueError("x and y must be the same size")，本质是买入/卖出信号的x轴数据（日期）和y轴数据（价格）长度不相等。合并绘制时完整数据集的覆盖效应，掩盖了单组信号的长度不匹配问题。

排查与修复步骤 #

1. 验证信号数据长度一致性
   直接打印买入、卖出信号的日期和价格数组长度，快速定位问题：

   print(len(buy_dates), len(buy_prices))
   print(len(sell_dates), len(sell_prices))
   

   若输出数值不等，说明数据筛选环节存在问题。

2. 修正信号数据筛选逻辑
   必须从同一行数据中提取日期和对应价格，避免跨数据集匹配导致的丢数：
   错误示例（易引发长度不匹配）：

   buy_dates = merged_data[merged_data['signal'] == 'buy']['date']
   buy_prices = stock_data[stock_data['date'].isin(buy_dates)]['low']
   

   正确示例（从同一数据集提取）：

   buy_signals = merged_data[merged_data['signal'] == 'buy']
   buy_dates = buy_signals['date']
   buy_prices = buy_signals['low']  # 替换为你需要标记的价格字段，如close/high
   

3. 清理空值或缺失数据
   排查并移除信号数据中的空值行：

   # 检查空值
   print(buy_signals[['date', 'low']].isnull().sum())
   print(sell_signals[['date', 'high']].isnull().sum())
   
   # 清理空值
   buy_signals = buy_signals.dropna(subset=['date', 'low'])
   sell_signals = sell_signals.dropna(subset=['date', 'high'])
   

修复后的绘制示例（以mplfinance为例） #

import mplfinance as mpf

# 生成买卖信号标记
buy_markers = mpf.make_addplot(buy_signals['low'], type='scatter', markersize=100, marker='^', color='g')
sell_markers = mpf.make_addplot(sell_signals['high'], type='scatter', markersize=100, marker='v', color='r')

# 绘制K线图+信号标记
mpf.plot(stock_data, type='candle', addplot=[buy_markers, sell_markers], figratio=(12,6))

额外注意事项 #

• 确保date字段为datetime类型，避免因格式不匹配导致的索引对齐错误
• 若使用其他绘图库（如Plotly），需保证x、y轴数据的长度严格一一对应

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