如何从 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()