matplotlib - 如何将两个标签添加到一个散点图中

Question

我有一个散点图。但是，我使用另一列（Playoffs - {0,1}）区分了这些点。

现在，如何为Playoffs赋予0值标签，1值为另一个标签？

fig = plt.figure(figsize=(8, 6))
plt.scatter(nba.ptsDiff, nba.W, s=25, c=nba.Playoffs, label='bla')
plt.legend()
plt.show()

Answer 1

尝试过滤列中的季后赛值。然后使用单独的plt.scatter命令绘制每个图。最后使用图例列表和图例中的图例标签。

fig = plt.figure(figsize=(8, 6))
nba1 = nba[nba['Playoffs'] == 1]
nba0 = nba[nba['Playoffs'] == 0]
pla = plt.scatter(nba1.ptsDiff, nba1.W, color='red')
plb = plt.scatter(nba0.ptsDiff, nba0.W, color='blue')

plt.legend([pla, plb], ["Attr A", "Attr B"])

plt.show()