来自 3D 数据的热图,带有浮点数

时间:2021-04-17 14:39:25

标签: python arrays numpy seaborn heatmap

我正在尝试从 csv 文件中的 3D 数据生成热图。 csv 文件的每行格式为 x,y,z。问题是当我创建一个数组来链接值时,我不能使用浮点数作为键。在 np.loadtext() 中将 dtype 设置为 int 时,代码工作正常;但这使得分辨率仅为 csv 文件可以复制的分辨率的一半。是否有另一种链接值的方法?

目前的代码是:

import numpy as np
import seaborn as sb
import matplotlib.pyplot as plt

fname = 'test18.csv'
x, y, z = np.loadtxt(fname, delimiter=',', dtype=float).T

pltZ = np.zeros((y.max()+1, x.max()+1), dtype=float)
pltZ[y, x] = z

heat_map = sb.heatmap(pltZ, cmap=plt.cm.rainbow)

plt.show()

0 个答案:

没有答案
相关问题