如何在streamlit中显示热图颜色相关图

import streamlit as st
import numpy as np
import seaborn as sns
import matplotlib.pyplot as plt

data = np.random.rand(10, 10)  # 生成一个10x10的随机数据矩阵

fig, ax = plt.subplots()
sns.heatmap(data, cmap='YlOrRd', annot=True, fmt=".2f", linewidths=0.5, ax=ax)

st.pyplot(fig)

import streamlit as st
import numpy as np
import seaborn as sns
import matplotlib.pyplot as plt

data = np.random.rand(10, 10)

fig, ax = plt.subplots()
sns.heatmap(data, cmap='YlOrRd', annot=True, fmt=".2f", linewidths=0.5, ax=ax)

st.pyplot(fig)