温馨提示×

如何使用Matplotlib制作热力图

小亿
115
2024-05-20 12:53:31
栏目: 编程语言

要制作热力图,可以使用Matplotlib中的imshow函数。下面是一个简单的示例代码,演示如何使用Matplotlib制作热力图:

import matplotlib.pyplot as plt
import numpy as np

# 创建一个随机的2D数组
data = np.random.rand(10, 10)

plt.imshow(data, cmap='hot', interpolation='nearest')
plt.colorbar()  # 添加颜色条
plt.show()

在这个示例中,我们首先生成一个随机的10x10的2D数组,然后使用imshow函数将该数组显示为热力图。参数cmap用于指定颜色映射,interpolation用于指定插值方法。最后使用colorbar函数添加颜色条,使得热力图更易于理解。运行代码后,将会显示出一个随机生成的热力图。

0