我正在使用Python和CSV文件。我正在尝试修改下面的散点图(2d),以根据csv文件中的第三列更改颜色。在搜索了多个帖子之后,我基本上想使用一个通用的colormap(rainbow)并将我的第三个数组乘以colormap,以便为每个xy点显示不同的颜色。我想我可以从ax.scatter
函数中做任何事情,但是我不知道如何将每个不同的x,y坐标乘以colormap和第三个数组号。它看起来应该类似于等高线图,但我更喜欢不同颜色的散点图。
下面是我使用的代码:
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 = ?);
看看散射线
它产生
相关问题 更多 >
编程相关推荐