使用Python matplotlib,我需要从一个csv文件构造一个热图,其中第一列和第二列分别具有x和y变量。第三列具有z变量(它是x和y的函数)。我想使用2D热图在颜色上表示z变量。
csv文件中的x和y变量没有特定顺序。
我开始研究一些代码以从文件中创建矩阵,但是没有继续进行。
def heat_map_simple(csv_file,x_var,y_var, color_var):
data = pd.read_csv(csv_file)
data = pd.DataFrame(data)
x_n = len(set(data.iloc[:][x_var]))
y_n = len(set(data.iloc[:][y_var]))
matrix = np.zeros((x_n, y_n))
rows = data.shape[1]
for r in xrange(rows):
val = data.iloc[r][color_var]