为MPLFinance图表添加第三个子图出现重叠问题求助

问题背景 #

需要构建三行全宽动态K线图表：顶部显示蜡烛图、中部显示成交量、底部显示技术指标（后续可扩展更多子图）。已实现带点击暂停/播放功能的MPLFinance动态图表，但遇到两个核心问题：

1. 初始代码存在子图重叠问题（成交量图底部被截断）
2. 更新代码修复重叠后，动画回放时图表无法随加载的数据自动扩展

运行环境 #

• Windows 10
• Python 3.10.8
• mplfinance==0.12.9b1
• pandas==1.4.3
• matplotlib==3.5.1

问题原因分析 #

1. 子图重叠：初始代码手动使用fig.add_subplot()时，子图网格参数（2,1,1、3,1,2、4,1,3）不匹配，导致面板空间分配混乱。
2. 图表无法扩展：更新代码中仅清空坐标轴重新绘图，但未触发坐标轴范围的自动调整，旧的x/y轴边界限制了新数据的展示。

修复后的完整代码 #

import mplfinance as mpf
import matplotlib.animation as animation
import pandas as pd

class ChartClass():
    def __init__(self, DF):
        self.DF = DF
        
        # 标准化列名并设置时间索引
        new_names = {
            'Open': 'open',
            'High': 'high',
            'Low': 'low',
            'Close': 'close',
            'Volume': 'volume',
        }
        self.DF.rename(columns=new_names, inplace=True)
        self.DF['Date'] = pd.to_datetime(self.DF['Date'])
        self.DF.set_index('Date', inplace=True)
        
        # 配置MPLFinance样式与面板布局
        s = mpf.make_mpf_style(base_mpf_style='charles',
                               rc={'figure.facecolor':'lightgray'})

        # 初始化图表，仅加载前20条数据作为初始显示
        self.fig, self.axes = mpf.plot(self.DF.iloc[:20],
                                       type='candle',
                                       figscale=1.5,
                                       figratio=(7,5),
                                       title='动态K线回放',
                                       style=s,
                                       volume=True,
                                       volume_panel=1,
                                       panel_ratios=(3,1,1),
                                       returnfig=True)
        
        # 提取各面板坐标轴
        self.ax_main = self.axes[0]
        self.ax_volu = self.axes[1]
        self.ax_ma = self.axes[2]
        
        # 手动初始化MA9指标线，避免重复创建
        self.ma_line, = self.ax_ma.plot([], [], color='lime', alpha=1)

        # 设置动画与暂停功能
        self.animation = animation.FuncAnimation(self.fig, self.update, interval=100)
        self.paused = False
        self.fig.canvas.mpl_connect('button_press_event', self.toggle_pause)
        mpf.show()
        
    def toggle_pause(self, *args, **kwargs):
        if self.paused:
            self.animation.resume()
        else:
            self.animation.pause()
        self.paused = not self.paused

    def update(self, ival):
        end_idx = 20 + ival
        if end_idx > len(self.DF):
            print('无更多数据可展示')
            self.animation.event_source.interval *= 3
            if self.animation.event_source.interval > 12000:
                exit()
            return
        
        # 获取当前要展示的数据段
        data = self.DF.iloc[:end_idx]
        
        # 清空并重新绘制蜡烛图与成交量
        self.ax_main.clear()
        self.ax_volu.clear()
        mpf.plot(data,
                 type='candle',
                 ax=self.ax_main,
                 volume=self.ax_volu,
                 style='charles',
                 show_nontrading=False)
        
        # 更新MA9指标线数据
        self.ma_line.set_data(data.index, data['MA9'])
        
        # 自动调整所有坐标轴范围，适配新数据
        self.ax_main.relim()
        self.ax_main.autoscale_view()
        self.ax_volu.relim()
        self.ax_volu.autoscale_view()
        self.ax_ma.relim()
        self.ax_ma.autoscale_view()

# 加载数据并添加MA9指标
df = pd.read_csv("./data/SPY5m.csv")
df['MA9'] = df['Close'].rolling(9).mean()

# 启动图表
ChartClass(DF=df)

修复要点说明 #

1. 原生面板布局：使用MPLFinance的panel_ratios参数分配三个面板的高度比例（3:1:1），避免手动创建子图的布局错误。
2. 高效指标更新：直接使用Matplotlib原生plot()方法创建指标线，替代每次调用mpf.make_addplot()，提升动画流畅度。
3. 自动适配坐标轴：每次更新后调用relim()和autoscale_view()，让坐标轴自动扩展以适配新增的数据。
4. 初始数据限制：初始化时仅加载前20条数据，保证动画从局部开始逐步扩展展示。

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