0

我使用从不同的 stackoverflow 线程中找到的代码从 csv 值的散点图创建了热图 在 MatPlotLib 中使用散点数据集生成热图

这可行，但我想编辑箱子之间的颜色/平滑度等。我已经阅读了这个https://matplotlib.org/examples/color/colormaps_reference.html ...但我的 n00b 水平正在阻止快速进展。我当前的代码是否似乎易于操作以在 bin 之间进行插值（平滑）或至少颜色变化，或者我是否需要以不同的方式创建热图以获得更多控制？（热图将根据跟踪项目的 xy 值表示及时使​​用空间的频率）谢谢，非常感谢任何帮助。

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()
4

0 回答 0