matplotlib中的圖例

我想知道如何執行以下操作：我有一個帶點和類的DataFrame。我想繪製所有點，併爲每個班級使用一種顏色。我如何指定類如何引用圖例中的顏色？matplotlib中的圖例

fig = plt.figure(figsize=(18,10), dpi=1600) 
df = pd.DataFrame(dict(points1 = data_plot[:,0], points2 = data_plot[:,1], \ 
      target = target[0:2000])) 
colors = {1: 'green', 2:'red', 3:'blue', 4:'yellow', 5:'orange', 6:'pink', \        
       7:'brown', 8:'black', 9:'white'} 
fig, ax = plt.subplots() 
ax.scatter(df['points1'], df['points2'], c = df['target'].apply(lambda x: colors[x]))

來源

2016-03-19 ivan

可以爲用戶提供的是你所得到的輸出和你想獲得輸出最小的可運行的例子嗎？這將更容易理解和回答你的問題。 –

，最簡單的辦法讓你的傳奇有單獨的條目爲每種顏色（因此它的target值）創建一個單獨的情節對象每個target值。

import matplotlib.pyplot as plt 
import pandas as pd 
import numpy as np 

x = np.random.rand(100) 
y = np.random.rand(100) 
target = np.random.randint(1,9, size=100) 

df = pd.DataFrame(dict(points1=x, points2=y, target=target)) 
colors = {1: 'green', 2:'red', 3:'blue', 4:'yellow', 5:'orange', 6:'pink', \ 
       7:'brown', 8:'black', 9:'white'} 
fig, ax = plt.subplots() 

for k,v in colors.items(): 
    series = df[df['target'] == k] 
    scat = ax.scatter(series['points1'], series['points2'], c=v, label=k) 

plt.legend()

來源

2016-03-19 17:07:53 Suever

我不會說'只'的方式，只是最直接的前進;）你可以用代理藝術家做一些有趣的事情。請參閱http://matplotlib.org/users/legend_guide.html#creating-artists-specifically-for-adding-to-the-legend-aka-proxy-artists – tacaswell

matplotlib中的圖例

回答

相關問題