可以在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
是带形状的numpy.array(7,2)。
这给了我一个这样的情节:
如果我更改了最后一行:
plt.scatter(coords[:, 0], coords[:, 1], marker='o', c=colors, s=50, edgecolor='None', label=range(simulations))
plt.legend(loc='lower left')
我明白了:
我想知道我是否必须做一个循环来进行分散并设置每个标签,如果有一些方法可以一次完成所有操作。
谢谢。
答案 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()