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

我最近偶然发现了一个类似的问题

这篇文章非常有用，我可以让各个脚本运行并为代码生成的随机测试数据创建热图。但是，我在修改代码以为正在处理的数据创建热图时遇到了问题。数据采用逗号分隔格式（.csv）

我目前在这个.csv文件中保存了3788对平均质量评级。这些平均质量评级都在0-5之间。我试图创建一个热图，在x轴和y轴（0-.499、.5-.999、1-1.499等）上以.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文件中读取数据，我将万分感激

Python有一个绝对壮观的csv文件库：

虽然我不知道matplotlib的详细信息，但下面的代码将在以制表符分隔的csv文件上循环，并在每个第一列和第二列上执行代码

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

with open(yourInputFile, "rb") as mycsv:
    reader = csv.DictReader(mycsv, dialect='excel-tab')

    for row in reader:
        x = row['name of first column']
        y = row['name of second column']
        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()

请注意，我使用的是DictReader变体，它需要一个标题才能工作。在文件的开头，或者作为构造函数的输入。您也可以使用常规的阅读器，但它使用的是列号而不是名称，这在100多个字段时非常麻烦

如有必要，您可以更改特定csv文件的方言，甚至创建自己的方言

---

最后，请注意，我对matplotlib一无所知，这段代码很可能是错误的。

因为您可以使用

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