我正在使用以下代码段,并且我想正确设置我的图例。以下是代码
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()
我需要完整的图例列表,例如blue
的{{1}},foo1
的{{1}}和green
的{{1}}。我该怎么办?