一行散点图标签 - Matplotlib

Question

是否可以在 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 = 7 和 coords 是形状为 (7, 2) 的 numpy.array。

这给了我这样的情节：

如果我更改最后一行：

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

我得到：

我想知道我是否必须做一个循环来进行分散并设置每个标签是否有某种方法可以一次完成。

谢谢。

Answer 1

我不知道如何用散点图来做。但是我不确定如果您想要不同的标签，使用 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()

一行散点图标签 - Matplotlib

Scatter plot labels in one line - Matplotlib

python

plot

matplotlib

legend