标签：matplotlib csv python

我最近偶然发现了一个关于how to generate a heatmap of frequencies in Python using the MatPlotLib module的类似问题.

这篇文章非常有用,我可以让各个脚本运行并为代码固有生成的随机测试数据创建热图.但是,我在修改代码以为正在使用的数据创建热图时遇到麻烦.数据采用逗号分隔格式(.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文件中读取数据,我将万分感谢！

解决方法:

由于您可以使用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() 

标签：matplotlib,csv,python
来源： https://codeday.me/bug/20191102/1993176.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。