Adicionar uma legenda à parcela do ponto Seaborn
estou a planear vários nomes de dados como enredo pontual usando seaborn
. Também estou a plotar todos os nomes de dados no mesmo eixo .
como acrescentaria uma lenda ao enredo ?
O meu código pega em cada um dos dados e enrola - os um após o outro na mesma figura.Cada nome de dados tem as mesmas colunas
date count
2017-01-01 35
2017-01-02 43
2017-01-03 12
2017-01-04 27
o meu código:
f, ax = plt.subplots(1, 1, figsize=figsize)
x_col='date'
y_col = 'count'
sns.pointplot(ax=ax,x=x_col,y=y_col,data=df_1,color='blue')
sns.pointplot(ax=ax,x=x_col,y=y_col,data=df_2,color='green')
sns.pointplot(ax=ax,x=x_col,y=y_col,data=df_3,color='red')
Este gráfico 3 linhas do mesmo enredo. No entanto, falta A Lenda. a documentação faz não aceitar label
argumento .
uma solução que funcionou foi criar um novo nome de dados e usar hue argument
.
df_1['region'] = 'A'
df_2['region'] = 'B'
df_3['region'] = 'C'
df = pd.concat([df_1,df_2,df_3])
sns.pointplot(ax=ax,x=x_col,y=y_col,data=df,hue='region')
Mas eu gostaria de saber se há uma maneira de criar uma lenda para o código que primeiro adiciona um enredo sequencial de pontos à figura e depois adiciona uma lenda.
saída da amostra:
11
Author: Spandan Brahmbhatt, 2017-03-13
1 answers
Sugiro que não use o seaborn para conspirar. Isto torna as coisas desnecessariamente complicadas.
Em vez disso, utilize matplotlib
No caso de ainda estar interessado em obter a legenda para pointplots, aqui uma maneira de ir:
Em vez disso, utilize matplotlib
plot_date
. Isto permite definir as legendas dos gráficos e colocá-las automaticamente numa legenda com ax.legend()
.
import matplotlib.pyplot as plt
import pandas as pd
import seaborn as sns
import numpy as np
date = pd.date_range("2017-03", freq="M", periods=15)
count = np.random.rand(15,4)
df1 = pd.DataFrame({"date":date, "count" : count[:,0]})
df2 = pd.DataFrame({"date":date, "count" : count[:,1]+0.7})
df3 = pd.DataFrame({"date":date, "count" : count[:,2]+2})
f, ax = plt.subplots(1, 1)
x_col='date'
y_col = 'count'
ax.plot_date(df1.date, df1["count"], color="blue", label="A", linestyle="-")
ax.plot_date(df2.date, df2["count"], color="red", label="B", linestyle="-")
ax.plot_date(df3.date, df3["count"], color="green", label="C", linestyle="-")
ax.legend()
plt.gcf().autofmt_xdate()
plt.show()
No caso de ainda estar interessado em obter a legenda para pointplots, aqui uma maneira de ir:
sns.pointplot(ax=ax,x=x_col,y=y_col,data=df1,color='blue')
sns.pointplot(ax=ax,x=x_col,y=y_col,data=df2,color='green')
sns.pointplot(ax=ax,x=x_col,y=y_col,data=df3,color='red')
ax.legend(handles=ax.lines[::len(df1)+1], labels=["A","B","C"])
ax.set_xticklabels([t.get_text().split("T")[0] for t in ax.get_xticklabels()])
plt.gcf().autofmt_xdate()
plt.show()
16
Author: ImportanceOfBeingErnest, 2017-03-13 17:45:47