从三个列表绘制热图X，Y，强度

Question

当我有x，y，强度时，我不知道如何创建热图（或等高线图）。我有一个这样的文件：

0,1,6
0,2,10
....

目前为止：

with open('eye_.txt', 'r') as f:
        for line in f:
                for word in line.split():
                        l = word.strip().split(',')
                        x.append(l[0])
                        y.append(l[1])
                        z.append(l[2])

尝试使用pcolormesh，但它需要一个shape对象，我不确定如何将这些列表转换为numpy数组。

我试过：

i,j = np.meshgrid(x,y)
arr = np.array(z)
plt.pcolormesh(i,j,arr)
plt.show()

它告诉我：

IndexError: too many indices

有人能阻止我用键盘猛击我的头吗？

Answer 1

如果没有规则网格（即每个网格值x和y的z值），则可以使用基于pandas数据帧的更通用的方法：

import pandas as pd
import seaborn as sns
import matplotlib.pypot as plt

data = pd.DataFrame(data={'x':x, 'y':y, 'z':z})
data = data.pivot(index='x', columns='y', values='z')
sns.heatmap(data)
plt.show()

pivot方法将使用索引和列中的唯一值来构造一个将缺少度量值设置为NaN的表。然后可以将表格绘制为热图。