Matplotlib 中生成熱圖
可以使用 Matplotlib 和 numpy 來建立熱圖。
熱圖示例
histogram2d 函式可以用來生成熱圖。
我們建立一些隨機資料陣列 (x, y) 以在程式中使用。我們將 bin 設定為 64，那熱圖將為 64×64。你可以更改 bins 數字，如果你想要另一個尺寸的話。
import numpy as np
import numpy.random
import matplotlib.pyplot as plt
# Create data
x = np.random.randn(4096)
y = np.random.randn(4096)
# Create heatmap
heatmap, xedges, yedges = np.histogram2d(x, y, bins=(64,64))
extent = [xedges[0], xedges[-1], yedges[0], yedges[-1]]
# Plot heatmap
plt.clf()
plt.title('Pythonspot.com heatmap example')
plt.ylabel('y')
plt.xlabel('x')
plt.imshow(heatmap, extent=extent)
plt.show()
結果：
)
此示例中的資料點是完全隨機的，使用 np.random.randn() 來生成。