如何從 python 中的一維數組創建二維熱圖？

Question

作為我自己的 function 的結果，我得到了類似於[0.4, 0.6, 0.2, 0.3, 0.9, 0.7]的一維數組

現在我想用它制作一個熱圖，例如，
0.7
0.3 0.9
0.4 0.6 0.2

或者

1 0.7 0.3 0.4
0.7 1 0.9 0.6
0.3 0.9 1 0.2
0.4 0.6 0.2 1
帶有一些適當的標簽。

我該怎么做呢？ 請幫我。

先感謝您。

Answer 1

您可以通過 python 中的 matplotlib 庫執行此操作。

import numpy as np
import matplotlib.pyplot as plt

data = [0.4, 0.6, 0.2, 0.3, 0.9, 0.7]

// reshape your data into 2D array
data = np.array(data).reshape(2, 3)

//  Now plot heatmap for your data show in graph
plt.imshow(data, cmap='hot')
plt.colorbar()
plt.show()