Python彩色热图不显示数据

时间:2015-12-28 17:18:59

标签: python matplotlib graph plot heatmap

遵循以下脚本:https://gist.github.com/teechap/9c066a9ab054cc322877我正在尝试从csv创建值的热图。数据读入很好并且创建了数组,甚至绘制了绘图但没有显示实际颜色。然而,据我所知,轴在正确的范围内。

import matplotlib.pyplot as plt
import numpy as np
from matplotlib.mlab import csv2rec

filein = open('test.csv', 'r')
data = csv2rec(filein, names=['x', 'y'])
x = data['x']
y = data['y']

intensity = np.random.randn(len(data['x']),len(data['y']))

x, y = np.meshgrid(x, y)

intensity = np.array(intensity)

plt.pcolormesh(x, y, intensity)
plt.colorbar() 
plt.show() 

test.csv看起来像:

3,4.32192809489
3,3.16992500144
3,2.32192809489
3,2.0
3,5.90689059561

我的意图是拥有更多的数据点,大约100,但正如您从下图中看到的那样,它实际上并不显示数据。

non-working-heatmap

1 个答案:

答案 0 :(得分:2)

我认为问题可能是你的所有x坐标都是一样的(3)。当我编辑输入值的第一列时,您的代码似乎对我很好。

您链接到的原始代码可以正常工作。