histogram2d热图操作

时间:2017-05-24 14:18:50

标签: numpy matplotlib heatmap histogram2d

我使用从不同的stackoverflow线程找到的代码在{csv值的散点图中创建了一个热图Generate a heatmap in MatPlotLib using a scatter data set

这有效,但我想编辑色块之间的颜色/平滑等。我已经阅读了这个https://matplotlib.org/examples/color/colormaps_reference.html ...但是我的n00b水平阻止了快速进展。我目前的代码是否可以轻松操作箱之间的插值(平滑)或至少颜色变化,或者我是否需要以不同的方式创建我的热图以获得更多控制? (热图将根据被跟踪项目的x y值表示空间在时间上的使用频率) 谢谢,任何帮助非常感谢。

import numpy as np
import matplotlib.pyplot as plt
import matplotlib as mpl


import csv

with open('myfile.csv') as csvfile:
     readCSV = csv.reader(csvfile, delimiter=',')
y = []
x = []
for row in readCSV:  
    x.append(float(row [0]))
    y.append(float(row [1]))

print (x, y)
heatmap, xedges, yedges = np.histogram2d(x,y,bins=20)
extent  = [xedges[0], xedges[-1], yedges[0], yedges[-1]]

plt.clf()
plt.imshow(heatmap.T, extent=extent)
plt.show()

0 个答案:

没有答案