散点图标签在一行 - Matplotlib

时间:2016-03-04 23:27:28

标签: python matplotlib plot legend

可以在Matplolib中一次设置所有标签吗?

例如,我有这段代码来绘制散点图:

cmap = plt.get_cmap('Set1')
colors = [cmap(i) for i in numpy.linspace(0, 1, simulations+1)]

plt.figure(figsize=(7, 7))
plt.scatter(coords[:, 0], coords[:, 1], marker='o', c=colors, s=50, edgecolor='None')
plt.legend(loc='lower left',)

其中simulations = 7coords是带形状的numpy.array(7,2)。

这给了我一个这样的情节:

enter image description here

如果我更改了最后一行:

plt.scatter(coords[:, 0], coords[:, 1], marker='o', c=colors, s=50, edgecolor='None', label=range(simulations))
plt.legend(loc='lower left')

我明白了:

enter image description here

我想知道我是否必须做一个循环来进行分散并设置每个标签,如果有一些方法可以一次完成所有操作。

谢谢。

1 个答案:

答案 0 :(得分:3)

我不确定如何使用散点图来做到这一点。但是,如果你想要不同的标签,我不确定使用scatter而不是plot是否有优势。

这个怎么样?

import numpy as np
import matplotlib.pyplot as plt

n = 10
coords = np.random.random((n,2))

cmap = plt.get_cmap('Set1')

for i, (x, y) in enumerate(coords):
    plt.plot(x, y, 'o', color=cmap(i/float(n)), label='%i'%i, ms=9, mec='none')

plt.axis((-0.5, 1.5, -0.5, 1.5))
plt.legend(loc='lower left', numpoints=1, frameon=False)
plt.show()

enter image description here