Seaborn regplot（Python 3）的自定義圖例

我一直在試圖關注這個How to make custom legend in matplotlib SO問題，但我認爲有些東西在翻譯中迷了路。我使用自定義顏色映射爲我的情節中的不同類別的點，我希望能夠將這些顏色標籤對放在一個表中。我將信息存儲在字典D_color_label中，然後製作了2個並行列表colors和labels。我試圖在ax.legend中使用它，但它似乎沒有工作。Seaborn regplot（Python 3）的自定義圖例

np.random.seed(0) 

# Create dataframe 
DF_0 = pd.DataFrame(np.random.random((100,2)), columns=["x","y"]) 

# Label to colors 
D_idx_color = {**dict(zip(range(0,25), ["#91FF61"]*25)), 
       **dict(zip(range(25,50), ["#BA61FF"]*25)), 
       **dict(zip(range(50,75), ["#916F61"]*25)), 
       **dict(zip(range(75,100), ["#BAF1FF"]*25))} 

D_color_label = {"#91FF61":"label_0", 
       "#BA61FF":"label_1", 
       "#916F61":"label_2", 
       "#BAF1FF":"label_3"} 

# Add color column 
DF_0["color"] = pd.Series(list(D_idx_color.values()), index=list(D_idx_color.keys())) 

# Plot 
fig, ax = plt.subplots(figsize=(8,8)) 
sns.regplot(data=DF_0, x="x", y="y", scatter_kws={"c":DF_0["color"]}, ax=ax) 

# Add custom legend 
colors = list(set(DF_0["color"])) 
labels = [D_color_label[x] for x in set(DF_0["color"])] 

# If I do this, I get the following error: 
# ax.legend(colors, labels) 
# UserWarning: Legend does not support '#BA61FF' instances. 
# A proxy artist may be used instead.

來源

2016-07-14 O.rka

根據http://matplotlib.org/users/legend_guide.html你必須把到將被標記的傳說功能的藝術家。要使用scatter_plot單獨您通過您的數據按顏色有組和一種顏色的每個數據繪製單獨設置自己的標籤每一個藝術家：

import pandas as pd 
import numpy as np 
import matplotlib.pylab as plt 
import seaborn as sns 

np.random.seed(0) 

# Create dataframe 
DF_0 = pd.DataFrame(np.random.random((100,2)), columns=["x","y"]) 
DF_0['color'] = ["#91FF61"]*25 + ["#BA61FF"]*25 + ["#91FF61"]*25 + ["#BA61FF"]*25 
#print DF_0 

D_color_label = {"#91FF61":"label_0", 
       "#BA61FF":"label_1", 
       "#916F61":"label_2", 
       "#BAF1FF":"label_3"} 
colors = list(set(DF_0["color"])) 
labels = [D_color_label[x] for x in set(DF_0["color"])] 

ax = sns.regplot(data=DF_0, x="x", y="y", scatter_kws={'c':DF_0['color'], 'zorder':1}) 

# Make a legend 
# groupby and plot points of one color 
ind = 0 
for i, grp in DF_0.groupby(['color']): 
    grp.plot(kind = 'scatter', x = 'x', y = 'y', c = i, ax = ax, label = labels[ind], zorder = 0) 
    ind += 1  
ax.legend(loc=2) 

plt.show()

來源

2016-07-14 22:03:10 Serenity

Seaborn regplot（Python 3）的自定義圖例

回答

相關問題