Matplotlib中怎么實(shí)現(xiàn)熱力圖

小億
110
2024-05-13 13:42:14
欄目: 編程語言

要在Matplotlib中繪制熱力圖，可以使用imshow()函數(shù)。以下是一個(gè)簡單的示例代碼：

import numpy as np
import matplotlib.pyplot as plt

data = np.random.rand(5, 5)  # 生成隨機(jī)數(shù)據(jù)

plt.imshow(data, cmap='hot', interpolation='nearest')  # 繪制熱力圖
plt.colorbar()  # 添加顏色條
plt.show()

在這個(gè)例子中，我們生成了一個(gè)隨機(jī)的5x5的數(shù)據(jù)矩陣，然后使用imshow()函數(shù)繪制熱力圖，指定使用hot顏色映射，并設(shè)置插值方法為nearest。最后通過colorbar()函數(shù)添加顏色條。運(yùn)行代碼后，將會(huì)顯示出一個(gè)熱力圖。

0 踩