要制作矩陣或二維數組的熱力圖,可以使用Matplotlib中的imshow函數。下面是一個簡單的示例代碼:
import numpy as np
import matplotlib.pyplot as plt
# 創建一個隨機的二維數組
data = np.random.rand(10, 10)
# 繪制熱力圖
plt.imshow(data, cmap='hot', interpolation='nearest')
plt.colorbar() # 添加顏色條
plt.show()
在這個示例中,我們首先使用numpy庫生成一個隨機的10x10的二維數組,然后使用imshow函數繪制熱力圖。參數cmap用于指定顏色映射,interpolation用于指定插值方法,plt.colorbar()用于添加顏色條。最后調用plt.show()顯示熱力圖。您也可以根據自己的需要調整數據以及參數來生成不同的熱力圖。