传奇中的标签不按顺序排列

时间:2014-05-11 16:02:34

标签: python matplotlib plot legend

我对图例中标签的顺序有一个奇怪的问题。我正在制作一个散点图,我在其中预先排序数据列表,以便在顶部绘制较小的点(在此问题中我提出了实现这一点的方法:Plot larger points on bottom and smaller on top

我遇到的问题是图例中标签的顺序显然是分散的。这是一个显示我的意思的MWE:

import matplotlib.pyplot as plt
import numpy as np

def rand_data():
    return np.random.uniform(low=0., high=1., size=(100,))

# Generate data.
x, y, x2, x3 = [rand_data() for i in range(4)]
# This data defines the markes and labels used.
x1 = np.random.random_integers(7, 9, size=(100,))

# Order all lists so smaller points are on top.
order = np.argsort(-np.array(x2))
# Order x and y.
x_o, y_o = np.take(x, order), np.take(y, order)
# Order list related to markers and labels.
z1 = np.take(x1, order)
# Order list related to sizes.
z2 = np.take(x2, order)
# Order list related to colors.
z3 = np.take(x3, order)

plt.figure()
cm = plt.cm.get_cmap('RdYlBu')

# Scatter plot where each value in z1 has a different marker and label
# assigned.
mrk = {7: ('o', '7'), 8: ('s', '8'), 9: ('D', '9')}
for key, value in mrk.items():

    s1 = (z1 == key)
    plt.scatter(x_o[s1], y_o[s1], marker=value[0], label=value[1],
        s=z2[s1] * 100., c=z3[s1], cmap=cm, lw=0.2)

# Plot colorbar
plt.colorbar()

# Plot legend.
plt.legend(loc="lower left", markerscale=0.7, scatterpoints=1, fontsize=10)

plt.show()

生成的图像如下所示:

enter image description here

可以看出,图例应显示8, 9, 7标签7, 8, 9

奇怪的是,如果我改为设置定义x1列表的行和mrk字典,其中包含标记和标签,如下所示:

x1 = np.random.random_integers(1, 3, size=(100,))
mrk = {1: ('o', '1'), 2: ('s', '2'), 3: ('D', '3')}

生成的图例 1, 2, 3排序。

那么,这里发生了什么,无论我在x1使用什么数字来定义标记和标签,我怎么能强制命令传奇?

1 个答案:

答案 0 :(得分:1)

嗯,那是因为dict .items()方法返回未排序的列表。但你可以这样做:

for key, value in sorted(mrk.items()):
    <..>

它将按要求的顺序获得(key, value)元组对:7 8 9

相关问题