У меня проблема с загрузкой данных CSV в matplotlib.
Вот так выглядит мой код:
import numpy as np
import matplotlib.cm as cm
import matplotlib.mlab as mlab
import matplotlib.pyplot as plt
import pandas as pd
csv_filename = 'heatmap_data.csv'
df = pd.read_csv(csv_filename)
delta = 0.050
x = y = np.arange(-3.0, 3.0, delta)
X, Y = np.meshgrid(x, y)
Z1 = mlab.bivariate_normal(X, Y, 1.0, 1.0, 0.0, 0.0)
Z2 = mlab.bivariate_normal(X, Y, 1.5, 0.5, 1, 1)
Z = Z2 - Z1 # difference of Gaussians
im = plt.imshow(Z, interpolation='bilinear', cmap=cm.bwr,
origin='lower', extent=[0, 100, 100, 0],
vmax=abs(Z).max(), vmin=-abs(Z).max())
cb = plt.colorbar()
plt.savefig("heatmap.png")
CSV Данные, которые я хочу загрузить:
x_pos,y_pos,type,importance
74,64,blue,-0.011517617893368
68,64,blue,-0.0041303878348102
32,64,red,0.049788810065569
8,64,red,0.12877712212094
88,64,red,0.0050599724578342
84,64,blue,-0.00052412736663743
80,64,blue,-0.020183850819375
78,64,blue,-0.01297132988303
72,64,red,0.080092605800612
64,64,red,0.074683215098353
62,64,blue,-0.011168199648943
58,64,red,0.030129086612831
x_pos - это координата X, y_pos - значение координаты Yэто что-то вроде координаты Z.Это означает важность точки.
Точки важности> 0 должны иметь красный цвет с акцентом на важность.
Точки важности <0 должны иметь синий цвет с акцентом на важность. </p>
Каждая координата на карте тепла без каких-либоданные должны быть белыми или прозрачными.Цели - это тепловая карта, аналогичная этому примеру образец тепловой карты .Возможно ли с этой библиотекой?