Matplotlib -显示二维数组的网格线和值？

import numpy as np
import matplotlib.pyplot as plt

# 创建一个二维数组
data = np.random.rand(10, 10)

# 绘制图表
plt.imshow(data, cmap='hot', interpolation='nearest')

# 显示网格线
plt.grid(True, which='both', color='black', linewidth=0.5)

# 显示数值
for i in range(data.shape[0]):
    for j in range(data.shape[1]):
        plt.text(j, i, f'{data[i, j]:.2f}', ha='center', va='center', color='white')

# 设置坐标轴
plt.xticks(range(data.shape[1]))
plt.yticks(range(data.shape[0]))

# 添加颜色条
plt.colorbar()

# 显示图表
plt.show()