У меня проблема с контурным графиком Matplotlib. У меня есть размер графика (100x100). Все значения важности CSV> 0 должны быть красного цвета. Каждое значение <0 должно быть синим. </p>
Мой вопрос: возможно ли иметь места без данных (без координат в CSV) белого цвета. Теперь он показывает какой-то средний цвет из близлежащих точек.
Это ситуация, которая у меня сейчас:
Это данные, отображаемые точками:
Этот тип графика мне нужен:
Можно ли получить подобный график с помощью Contour?
Это мой код:
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.mlab as ml
import pandas as pd
csv_filename = 'data.csv'
df = pd.read_csv(csv_filename)
data = [df.x_pos, df.y_pos]
x = df.x_pos
y = df.y_pos
z = df.importance
ny, nx = 1000, 1000
xmin, xmax = 0, 100
ymin, ymax = 0, 100
xi = np.linspace(xmin, xmax, nx)
yi = np.linspace(ymin, ymax, ny)
zi = ml.griddata(x, y, z, xi, yi, interp='linear')
plt.contour(xi, yi, zi, 1, linewidths = 0, colors = 'k')
plt.pcolormesh(xi, yi, zi, vmin=-1, vmax=1, cmap = plt.get_cmap('bwr'))
#plt.scatter(x, y, marker = 'o', c = z, s = 10, vmin=-1, vmax=1, cmap = plt.get_cmap('bwr'))
plt.colorbar()
plt.xlim(xmin, xmax)
plt.ylim(ymin, ymax)
plt.show()
** Это пример данных CSV: **
x_pos,y_pos,importance
32,-48,0.87487145766094
32,52,0.43743572883047
32,68,1.127350663193
Полные данные CSV здесь: https://pastebin.com/mYS6LSyG