溫馨提示×

Matplotlib怎么制作矩陣或二維數(shù)組的熱力圖

小億
119
2024-05-21 11:22:21
欄目: 編程語言

要制作矩陣或二維數(shù)組的熱力圖,可以使用Matplotlib中的imshow函數(shù)。下面是一個簡單的示例代碼:

import numpy as np
import matplotlib.pyplot as plt

# 創(chuàng)建一個隨機的二維數(shù)組
data = np.random.rand(10, 10)

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

在這個示例中,我們首先使用numpy庫生成一個隨機的10x10的二維數(shù)組,然后使用imshow函數(shù)繪制熱力圖。參數(shù)cmap用于指定顏色映射,interpolation用于指定插值方法,plt.colorbar()用于添加顏色條。最后調(diào)用plt.show()顯示熱力圖。您也可以根據(jù)自己的需要調(diào)整數(shù)據(jù)以及參數(shù)來生成不同的熱力圖。

0