基於分類類的matplotlib圖中的提及標簽

Question

我想在情節上貼上標簽。 我的代碼就像

import matplotlib.pyplot as plt
x1 = [1,2,3,4,5,6,7,8,9,10]
x2 = [7,3,2,4,1,4,3,5,8,4]
y  = [0,1,1,0,1,0,0,1,0,1]
plt.scatter(x1,x2,c=y)
plt.show()

輸出：

我想在圖上放置圖例標簽0和1 。 如果標簽是False或true，那么這樣做有好處嗎？

如果答案沒有任何迭代，將不勝感激

Answer 1

原則上， 帶有圖例的matplotlib散點圖或帶有圖例的 Matplotlib散點圖的問題說明了如何獲取散點圖的圖例。

由於您明確要求不要迭代，因此您當然可以用映射替換迭代。

import matplotlib.pyplot as plt
import numpy as np

x1 = [1,2,3,4,5,6,7,8,9,10]
x2 = [7,3,2,4,1,4,3,5,8,4]
y  = [0,1,1,0,1,0,0,1,0,1]
plt.scatter(x1,x2,c=y, cmap="viridis")

yunique = np.unique(y).astype(float)
handles = list(map(lambda i: plt.plot([], 
                color=plt.cm.viridis(i),marker="o", ls="")[0], yunique))
labels = list(map(str, yunique.astype(bool)))
plt.legend(handles=handles, labels=labels)

plt.show()

請注意，這只是一個純粹的學術示例，使用for循環要容易得多。