python - 如何在每个循环中更改图形的 xlabel

Question

下面的代码在每个循环中绘制了一个图形，我希望将每个矩阵的平均值打印为 x 标签。例如:ave 是 40。我不确定如何将每个图像的 ave 添加到 xlabel。

import matplotlib.pyplot as plt
import seaborn as sns
import numpy as np
a= np.random.randint(0, 100, size=(4, 600, 600))

for i in range(np.size(a,0)):
    b=a[i,:,:]

    ave=np.average(b)
   
    plt.figure()
    sns.heatmap(b, cmap='jet', square=True, xticklabels=False,
                yticklabels=False)
    plt.text(200,-20, "Relative Error", fontsize = 15, color='Black')
    plt.xlabel("ave is...")
    plt.show()

Answer 1

最好的方法是使用 F 字符串格式:

plt.xlabel(f'ave is {ave}')

请注意，为了避免数字有很多小数点，您可以使用

ave_round=np.round(ave, 3) # Round to 3 decimals

import matplotlib.pyplot as plt
import seaborn as sns
import numpy as np
a= np.random.randint(0, 100, size=(4, 600, 600))

for i in range(np.size(a,0)):
    b=a[i,:,:]

    ave=np.average(b)
    ave_round=np.round(ave, 3) # Round to 3 decimals
   
    plt.figure()
    sns.heatmap(b, cmap='jet', square=True, xticklabels=False,
                yticklabels=False)
    plt.text(200,-20, "Relative Error", fontsize = 15, color='Black')
    plt.xlabel(f"ave is {ave_round}")
    plt.show()