Python Matplotlib散点图:着色数据指向特定颜色

时间:2014-02-11 16:38:14

标签: python colors matplotlib scatter-plot

我正在尝试为特定颜色着色特定颜色,但没有让它起作用,任何人都可以帮助我吗?到目前为止,我有:

fig = plt.figure()

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]

scatter(x,y)

为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"}
fig = plt.figure()
ax = fig.add_subplot(111)
ax.scatter

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

但最后一行给出了一个关键错误:

Traceback (most recent call last):
  File "overallplot.py", line 91, in <module>
    ax.scatter(x,y,c=[use_colours[t[0]] for t in z],s=50)
KeyError: ' '

0 个答案:

没有答案