Я использую Python и файл CSV. В настоящее время я пытаюсь изменить график рассеяния (2d) ниже, чтобы изменить цвета на основе третьего столбца в моем CSV-файле. После поиска по нескольким сообщениям я в основном хочу использовать общую цветовую карту (радугу) и умножить мой третий массив на цветовую карту, чтобы отображать разные цвета для каждой из точек xy. Я думаю, что могу сделать все из функции ax.scatter
, но я не уверен, как умножить каждую другую координату x, y на цветовую карту и номер третьего массива. Он должен быть похож на контурный график, но я бы предпочел график рассеяния другого цвета.
Вот код, который я использую:
import matplotlib
from matplotlib.backends.backend_agg import FigureCanvasAgg as FigureCanvas
from matplotlib.figure import Figure
import matplotlib.mlab as mlab
import numpy as np
r = mlab.csv2rec('test.csv')
fig = Figure(figsize=(6,6))
canvas = FigureCanvas(fig)
ax = fig.add_subplot(111)
ax.set_title("X vs Y AVG",fontsize=14)
ax.set_xlabel("XAVG",fontsize=12)
ax.set_ylabel("YAVG",fontsize=12)
ax.grid(True,linestyle='-',color='0.75')
x = r.xavg #first column
y = r.yavg #second column
z = r.wtr #third column
ax.scatter(x,y,s=.2,c='b', marker = ',', cmap = ?);