如何根據散點圖中的顏色添加圖例[matplotlib]

Question

我一直試圖基於matplotlib中散點圖中的顏色添加圖例。 我已經看到了幾個示例，但是他們假設每個圖都將有一個與之關聯的標簽，並且我還沒有弄清楚如何在沒有“ label”屬性的情況下如何僅基於顏色來進行繪制。 目前我有以下代碼

plt.subplot(121)
plt.gca().set_title('Female')
survived = female_data[:,1]
pclass = female_data[:,2]
age = female_data[:,5]
label_color = ['r' if i==0 else 'b' for i in survived]
axes = plt.gca()
axes.get_xaxis().set_visible(False)
plt.scatter(age, pclass, color=label_color)

哪個渲染這樣的東西，我希望紅色和藍色在圖中有一個標簽。 任何想法？

Answer 1

https://colab.research.google.com/notebook#fileId=1RtdwW8ztZbBUpd67RWffKRVt0ae0dNkr

import matplotlib.pyplot as plt
import numpy as np

ages = np.array([1, 2, 1, 3, 1, 2, 2, 1, 1, 2])
pclass = np.arange(10)
survived = np.array([
    True, True, True, False, False,
    False, True, False, True, False
])
plt.plot(
    pclass[survived], ages[survived], 'bo',
    pclass[~survived], ages[~survived], 'ro'
)
plt.legend(['survived', 'died'])

希望您能看到我的colab筆記本或圖像。 這就是你想要的嗎？ 我不認為你要scatter 。

Answer 2

如果您以相同的順序將值的數組傳遞給圖例，則可以在散布中定義顏色，它應該可以工作。

標簽= ['0'，'其他'] plt.legend（標簽）