Matplotlib安装图例来自单独的专栏

时间:2018-08-06 14:42:30

标签: matplotlib plot scatter-plot

我正在使用以下代码段,并且我想正确设置我的图例。以下是代码

import numpy as np
import matplotlib.pyplot as plt

# Generate data...
x = np.random.random(10)
y = np.random.random(10)
z = np.random.choice(['foo1', 'foo2', 'foo3'], 10)

df = pd.DataFrame([x, y, z]).T 

colors = {'foo1': 'blue', 'foo2': 'green', 'foo3':'red'}
plt.scatter(x=df[0], y=df[1], marker='o', c= df[2].apply(lambda x: colors[x]))
plt.legend(colors)
plt.show()

enter image description here

我需要完整的图例列表,例如blue的{​​{1}},foo1的{​​{1}}和green的{​​{1}}。我该怎么办?

0 个答案:

没有答案