python - 如何从元组在python中生成热图？

Question

我有一个包含 n 个元素的元组，其中每个元素都是

([numpy.array([[r], [phi]]), E],)

其中 r 和 phi 是极坐标，E 是一些给定的实数值。如何从中构建热图？

标签： pythonheatmap

Answer 1

首先，将 n 个元素的元组拆分为列表列表：

# If data_tuple is the tuple of n elements
data_list = [[row[0][0][0][0], row[0][0][1][0], row[0][1]] for row in data_tuple]

然后使用seaborn构建热图

import seaborn as sns
sns.heatmap(data_list)

您可以在此处阅读有关 seaborn 热图的更多信息