matplotlib-如何向一个散点图添加两个标签

matplotlib-如何向一个散点图添加两个标签,matplotlib,Matplotlib,我有一个散点图。然而,我已经使用另一列(季后赛-{0,1})区分了分数 现在,我如何给季后赛的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() 尝试在列中过滤季后赛值。然后使用单独的plt.scatter命令绘制每个点。最后使用图例中的绘图列表和图例标签 fi

我有一个散点图。然而,我已经使用另一列(季后赛-{0,1})区分了分数

现在,我如何给季后赛的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()

尝试在列中过滤季后赛值。然后使用单独的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()

工作得很好。谢谢Keita