matplotlib - 1つの散布図に2つのラベルを追加するには

Question

私は散布図を1つ作成しました。しかし、別の列（Playoffs - {0、1}）を使ってポイントを区別しました。

ここで、Playoffsのラベルに0の値を、もう1つのラベルに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()