如何防止Matplotlib绘图挤压？IoT仪表板x轴扩展/滚动方案求助

针对IoT仪表板的多传感器数据动态绘图需求，你可以通过两种方式解决x轴挤压的问题：

方案1：x轴自动扩展 #

每次新增数据后，手动设置x轴范围，让x轴跟随最新数据自动延伸，避免数据被挤压。只需在绘图逻辑后添加x轴范围设置即可：

from matplotlib import animation, pyplot as plt
from itertools import count
import random

plt.style.use('fivethirtyeight')

x_vals = []
y_vals = []
index = count()

def animate(i):
    x_vals.append(next(index))
    y_vals.append(random.randint(0, 5))
    
    plt.cla()
    plt.scatter(x_vals, y_vals)
    plt.plot(x_vals, y_vals)
    # 让x轴最大值始终等于最新的x值，实现自动扩展
    plt.xlim(0, max(x_vals))

anime = animation.FuncAnimation(plt.gcf(), animate, interval=400)

plt.tight_layout()
plt.show()

方案2：滚动显示固定窗口数据 #

如果数据量持续增长，自动扩展x轴最终还是会导致图表过宽。这种情况下，适合用滚动窗口只显示最近的N个数据点（比如最近20个），保持图表始终清晰：

from matplotlib import animation, pyplot as plt
from itertools import count
import random

plt.style.use('fivethirtyeight')

x_vals = []
y_vals = []
index = count()
# 定义要显示的最新数据点数量
WINDOW_SIZE = 20

def animate(i):
    x_vals.append(next(index))
    y_vals.append(random.randint(0, 5))
    
    # 只保留最新的WINDOW_SIZE个数据，超出的从列表头部移除
    if len(x_vals) > WINDOW_SIZE:
        x_vals.pop(0)
        y_vals.pop(0)
    
    plt.cla()
    plt.scatter(x_vals, y_vals)
    plt.plot(x_vals, y_vals)
    # 设置x轴范围为当前窗口的x值区间
    plt.xlim(min(x_vals), max(x_vals))

anime = animation.FuncAnimation(plt.gcf(), animate, interval=400)

plt.tight_layout()
plt.show()

性能优化提示 #

频繁调用plt.cla()清除坐标轴会影响性能，尤其是大数据量场景。可以直接更新绘图元素的数据，避免重复创建：

from matplotlib import animation, pyplot as plt
from itertools import count
import random

plt.style.use('fivethirtyeight')

x_vals = []
y_vals = []
index = count()
WINDOW_SIZE = 20

# 初始化时创建线条和散点对象
line, = plt.plot([], [], '-')
scatter = plt.scatter([], [])
# 固定y轴范围，避免频繁变动
plt.ylim(-1, 6)

def animate(i):
    x_vals.append(next(index))
    y_vals.append(random.randint(0, 5))
    
    if len(x_vals) > WINDOW_SIZE:
        x_vals.pop(0)
        y_vals.pop(0)
    
    # 直接更新数据，无需清除坐标轴
    line.set_data(x_vals, y_vals)
    scatter.set_offsets(list(zip(x_vals, y_vals)))
    plt.xlim(min(x_vals), max(x_vals))
    
    return line, scatter

# 设置blit=True提升动画流畅度
anime = animation.FuncAnimation(plt.gcf(), animate, interval=400, blit=True)

plt.tight_layout()
plt.show()

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