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

Question

我最近偶然发现了一个类似的问题,关于如何使用MatPlotLib模块在Python中生成频率热图.

这篇文章非常有用,我可以让各个脚本运行并为代码本身生成的随机测试数据创建热图.但是,我无法调整代码来为我正在使用的数据创建热图.数据采用逗号描述格式(.csv).

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

我想将.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文件中的数据,我将永远感激不尽!

Answer 1

既然您可以numpy随意使用并假设您的csv文件表现良好,那么您可以使用numpy.loadtxt(),

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