如何在 streamlit 页面中为折线图制作动画

Question

我正在处理 ML 项目，我想（相对）实时显示具有适应度函数的图表。

我正在使用来自 this SO answer 的代码，只要图表显示在 matplotlib window 中，它就可以正常工作。一旦我把它添加到页面，它就变成了静态图像。

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.animation as animation

fig, ax = plt.subplots()

max_x = 5
max_rand = 10

x = np.arange(0, max_x)
ax.set_ylim(0, max_rand)
line, = ax.plot(x, np.random.randint(0, max_rand, max_x))


def init():  # give a clean slate to start
    line.set_ydata([np.nan] * len(x))
    return line,


def animate(i):  # update the y values (every 1000ms)
    line.set_ydata(np.random.randint(0, max_rand, max_x))
    return line,

ani = animation.FuncAnimation(
    fig, animate, init_func=init, interval=1000, blit=True, save_count=10)

st.pyplot(plt)

知道如何制作图表动画吗？它不一定是 matplotlib。

Answer 1

我收到了 answer on the streamlit forum 所以我只是在这里复制更新的代码

import matplotlib.pyplot as plt
import numpy as np
import streamlit as st
import time

fig, ax = plt.subplots()

max_x = 5
max_rand = 10

x = np.arange(0, max_x)
ax.set_ylim(0, max_rand)
line, = ax.plot(x, np.random.randint(0, max_rand, max_x))
the_plot = st.pyplot(plt)

def init():  # give a clean slate to start
    line.set_ydata([np.nan] * len(x))

def animate(i):  # update the y values (every 1000ms)
    line.set_ydata(np.random.randint(0, max_rand, max_x))
    the_plot.pyplot(plt)

init()
for i in range(100):
    animate(i)
    time.sleep(0.1)

如何在 streamlit 页面中为折线图制作动画

How to animate a line chart in a streamlit page

python

matplotlib

streamlit