Matplotlib实时动画中如何让x轴随数据动态更新取值范围

问题根源 #

x轴无法更新的核心原因是调用FuncAnimation时设置了blit=True，该模式下动画只会重绘update_line函数返回的折线对象，坐标轴范围、刻度这类不属于返回值的元素修改后不会被刷新到画布上。

修复方案 #

最简单的修复方式是将blit参数设置为False，同时建议改用Matplotlib面向对象的API操作坐标轴，避免全局plt调用的潜在问题。

代码优化说明 #

• 原DataSource类中的data、display是类属性，多实例化时会共享数据，改为在__init__中定义为实例属性更合理
• 简化DataSource的逻辑，无需多次判断列表长度，直接取切片即可实现取最新N条数据的效果
• 可按需新增y轴自适应逻辑，让y轴也随实时数据的范围动态调整

修改后的完整代码 #

DataSource类

import numpy as np

class DataSource:
    def __init__(self, display=10):
        self.data = []
        self.display = display

    # 追加随机数并返回最新display条数据
    def get_data(self):
        self.data.append(np.random.rand(1)[0])
        return self.data[-self.display:]

    # 返回最新display条数据对应的索引
    def get_index_vector(self):
        data_len = len(self.data)
        start = max(0, data_len - self.display)
        return list(range(start, data_len))

动画绘制代码

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.animation as animation
# 若DataSource定义在单独的datasource.py文件中保留该导入，否则可删除
from datasource import DataSource

def update_line(num, source, line, ax):
    data = source.get_data()
    indexs = source.get_index_vector()
    # 动态更新x轴范围
    ax.set_xlim(indexs[0], indexs[-1])
    # 如需整数刻度可取消下一行注释
    # ax.set_xticks(np.arange(indexs[0], indexs[-1]+1, 1))
    # 如需y轴自适应可取消下一行注释
    # ax.set_ylim(min(data)-0.1, max(data)+0.1)
    line.set_data(indexs, data)
    return line,

def plot_animation():
    fig1, ax = plt.subplots()
    source = DataSource()

    l, = ax.plot([], [], 'r-')

    ax.set_xlim(0, 10)
    ax.set_ylim(0, 1)
    ax.set_xlabel('x')
    ax.set_title('test')
    # 核心修改：blit设为False，同时传入ax作为参数
    line_ani = animation.FuncAnimation(fig1, update_line, fargs=(source, l, ax),
                                    interval=150, blit=False)
    plt.show()

if __name__ == "__main__":
    plot_animation()

内容的提问来源于stack exchange，提问作者Ferran Capallera Guirado