Adicionar legenda ao gráfico de linha ggplot2

143

Eu tenho uma pergunta sobre lendas no ggplot2. Consegui traçar três linhas no mesmo gráfico e quero adicionar uma legenda com as três cores usadas. Este é o código usado

library(ggplot2)    
require(RCurl)

link<-getURL("https://dl.dropbox.com/s/ds5zp9jonznpuwb/dat.txt")
datos<- read.csv(textConnection(link),header=TRUE,sep=";")
datos$fecha <- as.POSIXct(datos[,1], format="%d/%m/%Y")    

temp = ggplot(data=datos,aes(x=fecha, y=TempMax,colour="1")) + 
           geom_line(colour="red") + opts(title="TITULO") +
           ylab("Temperatura (C)") + xlab(" ") + 
           scale_y_continuous(limits = c(-10,40)) + 
           geom_line(aes(x=fecha, y=TempMedia,colour="2"),colour="green") + 
           geom_line(aes(x=fecha, y=TempMin,colour="2"),colour="blue") +
           scale_colour_manual(values=c("red","green","blue"))

temp

e a saída

ggplot três linhas

Gostaria de adicionar uma legenda com as três cores usadas e o nome da variável (TempMax, TempMedia e TempMin). eu tentei

scale_colour_manual

mas não consegue encontrar o caminho exato.

Infelizmente, os dados originais foram excluídos do site vinculado e não puderam ser recuperados. Mas eles vieram de arquivos de dados meteo com este formato

"date","Tmax","Tmin","Tmed","Precip.diaria","Wmax","Wmed"
2000-07-31 00:00:00,-1.7,-1.7,-1.7,-99.9,20.4,20.4
2000-08-01 00:00:00,22.9,19,21.11,-99.9,6.3,2.83
2000-08-03 00:00:00,24.8,12.3,19.23,-99.9,6.8,3.87
2000-08-04 00:00:00,20.3,9.4,14.4,-99.9,8.3,5.29
2000-08-08 00:00:00,25.7,14.4,19.5,-99.9,7.9,3.22
2000-08-09 00:00:00,29.8,16.2,22.14,-99.9,8.5,3.27
2000-08-10 00:00:00,30,17.8,23.5,-99.9,7.7,3.61
2000-08-11 00:00:00,27.5,17,22.68,-99.9,8.8,3.85
2000-08-12 00:00:00,24,13.3,17.32,-99.9,8.4,3.49
pacomet
fonte
Ainda estou curioso para saber se as lendas podem ser vinculadas a elementos separados do gráfico (como geom_line diferente).
Etienne Low-Décarie
Se você tiver apenas três linhas, sugiro olhar para o pacote dirrectlabels. (LINK)
Tyler Rinker
@TylerRinker eu tinha usado antes para outros fins, mas agora a resposta de csgillespie funciona melhor para mim
pacomet
@ EtienneLow-Décarie Você pode, mas em geral apenas se eles usarem estética diferente. por exemplo, mapeando um conjunto de linhas para cores e outro para tipo de linha. Normalmente, você também passaria dados separados para cada geom nesse caso.
joran

Respostas:

82

Costumo descobrir que, se estou especificando cores individuais em vários geom's, estou fazendo errado. Aqui está como eu plotaria seus dados:

##Subset the necessary columns
dd_sub = datos[,c(20, 2,3,5)]
##Then rearrange your data frame
library(reshape2)
dd = melt(dd_sub, id=c("fecha"))

Tudo o que resta é um simples comando ggplot:

ggplot(dd) + geom_line(aes(x=fecha, y=value, colour=variable)) +
  scale_colour_manual(values=c("red","green","blue"))

Gráfico de exemplo

insira a descrição da imagem aqui

csgillespie
fonte
87
Ainda estou curioso para saber como adicionar legendas associadas à adição separada de elementos como geom_line, que, embora eu fosse o objetivo original da pergunta.
Etienne Low-Décarie
201

Como o @Etienne perguntou como fazer isso sem derreter os dados (que geralmente é o método preferido, mas reconheço que pode haver alguns casos em que isso não é possível), apresento a seguinte alternativa.

Comece com um subconjunto dos dados originais:

datos <-
structure(list(fecha = structure(c(1317452400, 1317538800, 1317625200, 
1317711600, 1317798000, 1317884400, 1317970800, 1318057200, 1318143600, 
1318230000, 1318316400, 1318402800, 1318489200, 1318575600, 1318662000, 
1318748400, 1318834800, 1318921200, 1319007600, 1319094000), class = c("POSIXct", 
"POSIXt"), tzone = ""), TempMax = c(26.58, 27.78, 27.9, 27.44, 
30.9, 30.44, 27.57, 25.71, 25.98, 26.84, 33.58, 30.7, 31.3, 27.18, 
26.58, 26.18, 25.19, 24.19, 27.65, 23.92), TempMedia = c(22.88, 
22.87, 22.41, 21.63, 22.43, 22.29, 21.89, 20.52, 19.71, 20.73, 
23.51, 23.13, 22.95, 21.95, 21.91, 20.72, 20.45, 19.42, 19.97, 
19.61), TempMin = c(19.34, 19.14, 18.34, 17.49, 16.75, 16.75, 
16.88, 16.82, 14.82, 16.01, 16.88, 17.55, 16.75, 17.22, 19.01, 
16.95, 17.55, 15.21, 14.22, 16.42)), .Names = c("fecha", "TempMax", 
"TempMedia", "TempMin"), row.names = c(NA, 20L), class = "data.frame")

Você pode obter o efeito desejado por (e isso também limpa o código de plotagem original):

ggplot(data = datos, aes(x = fecha)) +
  geom_line(aes(y = TempMax, colour = "TempMax")) +
  geom_line(aes(y = TempMedia, colour = "TempMedia")) +
  geom_line(aes(y = TempMin, colour = "TempMin")) +
  scale_colour_manual("", 
                      breaks = c("TempMax", "TempMedia", "TempMin"),
                      values = c("red", "green", "blue")) +
  xlab(" ") +
  scale_y_continuous("Temperatura (C)", limits = c(-10,40)) + 
  labs(title="TITULO")

A idéia é que cada linha receba uma cor mapeando a colourestética para uma sequência constante. Escolher a sequência que é o que você deseja que apareça na legenda é a mais fácil. O fato de que, nesse caso, é o mesmo que o nome da yvariável que está sendo plotada não é significativo; poderia ser qualquer conjunto de strings. É muito importante que isso esteja dentro da aeschamada; você está criando um mapeamento para essa "variável".

scale_colour_manualagora pode mapear essas seqüências de caracteres para as cores apropriadas. O resultado é insira a descrição da imagem aqui

Em alguns casos, o mapeamento entre os níveis e as cores precisa ser explicitado, nomeando os valores na escala manual (obrigado a @DaveRGP por apontar isso):

ggplot(data = datos, aes(x = fecha)) +
  geom_line(aes(y = TempMax, colour = "TempMax")) +
  geom_line(aes(y = TempMedia, colour = "TempMedia")) +
  geom_line(aes(y = TempMin, colour = "TempMin")) +
  scale_colour_manual("", 
                      values = c("TempMedia"="green", "TempMax"="red", 
                                 "TempMin"="blue")) +
  xlab(" ") +
  scale_y_continuous("Temperatura (C)", limits = c(-10,40)) + 
  labs(title="TITULO")

(fornecendo a mesma figura de antes). Com valores nomeados, as quebras podem ser usadas para definir a ordem na legenda e qualquer ordem pode ser usada nos valores.

ggplot(data = datos, aes(x = fecha)) +
  geom_line(aes(y = TempMax, colour = "TempMax")) +
  geom_line(aes(y = TempMedia, colour = "TempMedia")) +
  geom_line(aes(y = TempMin, colour = "TempMin")) +
  scale_colour_manual("", 
                      breaks = c("TempMedia", "TempMax", "TempMin"),
                      values = c("TempMedia"="green", "TempMax"="red", 
                                 "TempMin"="blue")) +
  xlab(" ") +
  scale_y_continuous("Temperatura (C)", limits = c(-10,40)) + 
  labs(title="TITULO")

Brian Diggs
fonte
2
Eu amo essa solução, mas acho que pode haver uma limitação. Existe um problema de classificação alfabética entre o mapeamento das variáveis ​​'quebras' e 'valores'? TempM {a] x, TempM {e} dia e TempM {i} n são ordenados, embora quando eu adapto isso aos meus nomes de variáveis, as cores parecem corresponder em ordem alfabética às 'quebras', não na entrada de ordem . Os itens acima podem ser esclarecidos / refinados para refletir / corrigir isso?
precisa saber é o seguinte
3
Consegui encontrar uma correção para o problema que comprei anteriormente: re: color order. use o formulário em scale_colour_manual("", values = c("TempMax" = "red", "TempMedia" = "green", "TempMin" = "blue"))que TempMax, TempMedia e TempMin são especificados como argumento de cores, como na resposta acima.
DaveRGP
@DaveRGP Poderia ser considerado um bug do ggplot?
Alessandro Jacopson 16/06
1
@StellaBiderman Obrigado. É bom saber que essa resposta ainda é útil (quase) 5 anos (!) Depois.
Brian Diggs
1
@BrianDiggs Você não saberia como fazer isso mostrar um ponto na escala em oposição a uma linha, não é?
Stella Biderman
2

Eu realmente gosto da solução proposta por @Brian Diggs. No entanto, no meu caso, crio os gráficos de linha em um loop, em vez de explicitamente, porque não sei a priori quantos gráficos terei. Quando tentei adaptar o código do @ Brian, tive alguns problemas ao lidar com as cores corretamente. Aconteceu que eu precisava modificar as funções estéticas. Caso alguém tenha o mesmo problema, aqui está o código que funcionou para mim.

Eu usei o mesmo quadro de dados que @Brian:

data <- structure(list(month = structure(c(1317452400, 1317538800, 1317625200, 1317711600, 
                                       1317798000, 1317884400, 1317970800, 1318057200, 
                                       1318143600, 1318230000, 1318316400, 1318402800, 
                                       1318489200, 1318575600, 1318662000, 1318748400, 
                                       1318834800, 1318921200, 1319007600, 1319094000), 
                                     class = c("POSIXct", "POSIXt"), tzone = ""),
                   TempMax = c(26.58, 27.78, 27.9, 27.44, 30.9, 30.44, 27.57, 25.71, 
                               25.98, 26.84, 33.58, 30.7, 31.3, 27.18, 26.58, 26.18, 
                               25.19, 24.19, 27.65, 23.92), 
                   TempMed = c(22.88, 22.87, 22.41, 21.63, 22.43, 22.29, 21.89, 20.52,
                                 19.71, 20.73, 23.51, 23.13, 22.95, 21.95, 21.91, 20.72, 
                                 20.45, 19.42, 19.97, 19.61), 
                   TempMin = c(19.34, 19.14, 18.34, 17.49, 16.75, 16.75, 16.88, 16.82, 
                               14.82, 16.01, 16.88, 17.55, 16.75, 17.22, 19.01, 16.95, 
                               17.55, 15.21, 14.22, 16.42)), 
              .Names = c("month", "TempMax", "TempMed", "TempMin"), 
              row.names = c(NA, 20L), class = "data.frame")  

No meu caso, eu giro my.colse my.namesdinamicamente, mas não quero tornar as coisas desnecessariamente complicadas, por isso dou explicitamente aqui. Essas três linhas facilitam a ordenação da legenda e a atribuição de cores.

my.cols <- heat.colors(3, alpha=1)
my.names <- c("TempMin", "TempMed", "TempMax")
names(my.cols) <- my.names

E aqui está o enredo:

p <-  ggplot(data, aes(x = month))

for (i in 1:3){
  p <- p + geom_line(aes_(y = as.name(names(data[i+1])), colour = 
colnames(data[i+1])))#as.character(my.names[i])))
}
p + scale_colour_manual("", 
                        breaks = as.character(my.names),
                        values = my.cols)
p

insira a descrição da imagem aqui

Justyna
fonte
2
Com essa complexidade, fica realmente muito mais fácil remodelar seus dados no formato longo ggplotesperado.
Axeman
1
Eu não acho que realmente acrescente complexidade à comparação com a resposta original postada por @Brian. Além disso, algumas pessoas podem querer fazer isso sem remodelar os dados.
Justyna
... e esta abordagem permite Geoms diferentes (tipos de enredo) pela variável
mac