溫馨提示×

如何使用Matplotlib制作熱力圖

小億
115
2024-05-20 12:53:31
欄目: 編程語言

要制作熱力圖,可以使用Matplotlib中的imshow函數(shù)。下面是一個簡單的示例代碼,演示如何使用Matplotlib制作熱力圖:

import matplotlib.pyplot as plt
import numpy as np

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

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

在這個示例中,我們首先生成一個隨機的10x10的2D數(shù)組,然后使用imshow函數(shù)將該數(shù)組顯示為熱力圖。參數(shù)cmap用于指定顏色映射,interpolation用于指定插值方法。最后使用colorbar函數(shù)添加顏色條,使得熱力圖更易于理解。運行代碼后,將會顯示出一個隨機生成的熱力圖。

0