在Python MatPlotLib中生成频率热图,从.csv文件读取X和Y坐标

这篇文章非常有用,我可以让各个脚本运行并为代码固有生成的随机测试数据创建热图.但是,我在修改代码以为正在使用的数据创建热图时遇到麻烦.数据采用逗号分隔格式(.csv).

我目前在此.csv文件中保存了3788对平均质量评级.这些平均质量评级的范围均为0-5.我正在尝试创建一个热图,以x和y轴(0-.499,.5-.999、1-1.499)按0.5增量对数据进行分档等等).

我想将.csv文件的第一列(webqualityratings)导入为热图的x值,并将.csv文件的第二列(inpersonqualityratings)导入为热图的y值.

我试图改编的“ ptomato”发布的代码,并由Mike Graham编辑,如下所示：

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

# Generate some test data  
x = np.random.randn(8873)  
y = np.random.randn(8873)  

heatmap, xedges, yedges = np.histogram2d(x, y, bins=50)  
extent = [xedges[0], xedges[-1], yedges[0], yedges[-1]]  
plt.clf()  
plt.imshow(heatmap, extent=extent)  
plt.show() 

如果有人可以帮助我修改此代码以按照指定的方式从我的.csv文件中读取数据,我将万分感谢！

import numpy as np   
import matplotlib.pyplot as plt  

dat = np.loadtxt('mydata.csv')

x, y = dat[:,0], dat[:,1]

heatmap, xedges, yedges = np.histogram2d(x, y, bins=50)  
extent = [xedges[0], xedges[-1], yedges[0], yedges[-1]]  
plt.clf()  
plt.imshow(heatmap, extent=extent)  
plt.show()