使用python在散点图中着色数据点

时间:2014-02-11 15:30:44

标签: python colors matplotlib scatter-plot

我正在尝试使用某些颜色为某些数据点绘制散点图。

data = np.genfromtxt('overallplotfile.csv',delimiter=',',dtype=str,skiprows=1)
data2 = np.genfromtxt('overallplotfile.csv',delimiter=',',dtype=float,skiprows=1)
x = data2[:,1]
y = data2[:,3]
z = data[:,2]

第2列(z)现在包含我想稍后为其指定颜色的电台名称。

scatter(x,y)
plot(x,yp)

为z中的电台名称指定颜色。

    use_colours = {"KNZ": "red", "PXZ": "red", "BFZ": "red","MQZ": "red","OPZ": "red","TUZ": "red","PUZ": "red","TSZ": "red","WEL": "red","MRZ": "red","KHZ": "red","ODZ": "red","MLZ": "red",    "VRZ": "green", "WIZ": "green", "NNZ": "green",  "THZ": "green",  "WVZ": "green",  "WKZ": "green", "MLZ": "green","WHZ": "green",  "PYZ": "green", "TOZ": "blue","TLZ": "cyan", "HIZ": "cyan","QRZ": "cyan","DSZ": "cyan", "FOZ": "cyan","JCZ": "cyan","MSZ": "cyan","DCZ": "cyan",  "WAZ": "yellow",  "WIZ": "yellow",  "URZ": "yellow","LTZ": "yellow","RPZ": "yellow","LBZ": "yellow","EAZ": "yellow", "MXZ" = "magenta", "BKZ" = "magenta","MWZ" = "magenta", "OXZ" = "magenta", "APZ" = "magenta"}

    ax.scatter(x,y,c=[use_colours[x[0]] for x in z],s=50)

使用电台名称标记数据点。     地图(文本,X,Y,Z)

但是,当我尝试将颜色分配给工作站时,我收到无效的语法错误。任何人都可以帮助我吗?

1 个答案:

答案 0 :(得分:1)

您在=定义的末尾使用:符号而不是dict符号。这可能会导致SyntaxError