Como adicionar uma legenda ggplot2 com tamanho e cor diferentes?

89

Estou usando o ggplot2 para melhorar os gráficos de barras de precipitação.

Aqui está um exemplo reproduzível do que desejo alcançar:

library(ggplot2)
library(gridExtra)
secu <- seq(1, 16, by=2)
melt.d <- data.frame(y=secu, x=LETTERS[1:8])
m <- ggplot(melt.d, aes(x=x, y=y)) +
  geom_bar(fill="darkblue") + 
  labs(x="Weather    stations", y="Accumulated Rainfall [mm]") +
  opts(axis.text.x=theme_text(angle=-45, hjust=0, vjust=1),
       title=expression("Rainfall"), plot.margin = unit(c(1.5, 1, 1, 1), "cm"),
       plot.title = theme_text(size = 25, face = "bold", colour = "black", vjust = 5))
z <- arrangeGrob(m, sub = textGrob("Location", x = 0, hjust = -3.5, vjust = -33, gp = gpar(fontsize = 18, col = "gray40"))) #Or guessing x and y with just option
z

Não sei como evitar o uso de adivinhação de números em hjust e vjust em ggplot2. Existe uma maneira melhor de colocar uma legenda (não apenas usando \ n, mas uma legenda com cor e tamanho de texto diferente)?

Eu preciso poder usar o ggsave para ter um arquivo pdf.

Aqui estão duas questões relacionadas:

Adicionar uma citação de nota de rodapé fora da área do gráfico em R?

Como posso adicionar uma legenda e alterar o tamanho da fonte dos gráficos do ggplot em R?

Obrigado por qualquer ajuda.

Migue
fonte
O vjust = -33 funcionou para mim no Linux. Eu sei que o sub é feito para ir abaixo do enredo, mas foi a única maneira de conseguir o que queria.
Migue de
por alguma razão, isso torna meu enredo muito pequeno e cria um espaço enorme abaixo do gráfico
zazu 01 de
2
A resposta de @hrbrmstr parece ser o caminho a percorrer hoje em dia
andschar 01 de

Respostas:

103

As versões mais recentes do ggplot2 (ou seja, 2.1.0.9000 ou mais recente) têm legendas e legendas abaixo do gráfico como funcionalidade incorporada. Isso significa que você pode fazer isso:

library(ggplot2) # 2.1.0.9000+ 

secu <- seq(1, 16, by=2)
melt.d <- data.frame(y=secu, x=LETTERS[1:8])

m <-  ggplot(melt.d, aes(x=x, y=y))
m <- m + geom_bar(fill="darkblue", stat="identity")
m <- m + labs(x="Weather    stations", 
              y="Accumulated Rainfall [mm]",
              title="Rainfall",
              subtitle="Location")
m <- m + theme(axis.text.x=element_text(angle=-45, hjust=0, vjust=1)) 
m <- m + theme(plot.title=element_text(size=25, hjust=0.5, face="bold", colour="maroon", vjust=-1))
m <- m + theme(plot.subtitle=element_text(size=18, hjust=0.5, face="italic", color="black"))
m
hrbrmstr
fonte
Erro em (função (el, elname): "plot.subtitle" não é um nome de elemento de tema válido.
Luís de Sousa
1
Leia as respostas: "últimas compilações do ggplot2 (ou seja, 2.1.0.9000 ou mais recente)"
hrbrmstr
75

Ignorar esta resposta A ggplot2 versão 2.2.0 tem funcionalidade de título e subtítulo. Veja a resposta de @hrbrmstr abaixo .


Você pode usar atopfunções aninhadas dentro de um expressionpara obter tamanhos diferentes.

EDITAR código atualizado para ggplot2 0.9.3

m <-  ggplot(melt.d, aes(x=x, y=y)) + 
     geom_bar(fill="darkblue", stat = "identity") + 
     labs(x="Weather    stations", y="Accumulated Rainfall [mm]") + 
     ggtitle(expression(atop("Rainfall", atop(italic("Location"), "")))) +
     theme(axis.text.x = element_text(angle=-45, hjust=0, vjust=1), 
     #plot.margin = unit(c(1.5, 1, 1, 1), "cm"), 
     plot.title = element_text(size = 25, face = "bold", colour = "black", vjust = -1))

insira a descrição da imagem aqui

Sandy Muspratt
fonte
10
Olá, esta é uma solução incrível. Eu gostaria de usá-lo, mas em vez do atop(italic("Location")que eu gostaria de ter um objeto: atop(italic(my_string_vector). Eu tentei isso, mas então a legenda foi avaliada para (my_string_vector) . Como forçar esta expressão a usar o valor da string e não tratar o texto fornecido literalmente?
Konrad
1
caso você esteja tendo problemas ao usar variáveis ​​que você deve usar em bquotevez de expression, veja aqui
toto_tico
3
@ Konrad usar objetos, substitua expressioncom bquotee enrole os objetos com .(), assim, para um título principal armazenado em um objeto chamado "main.title" e para um subtítulo armazenados em um objeto chamado "sub.title": ggtitle(bquote(atop(.(main.title), atop(italic(.(sub.title)), "")))) O crédito vai para Didzis A resposta de Elferts aqui: stackoverflow.com/questions/19957536/…
coip
10

Parece que optsestá obsoleto a partir do ggplot 2 0.9.1 e não é mais funcional. Isso funcionou para mim com as versões mais recentes a partir de hoje: + ggtitle(expression(atop("Top line", atop(italic("2nd line"), "")))).

Aren Cambre
fonte
Isso também funciona sem o final , ""- para que serve essa parte?
naught101
Me bate. Posso ter copiado um exemplo que vi em outro lugar.
Aren Cambre
Provavelmente da resposta de @SenyMuspratt acima: P - Eu entendo agora, atop()é algo como uma fração sem barras. Portanto, colocar o segundo atop()dentro do primeiro fornece uma subfração, com o texto proporcionalmente menor. O ""é a parte inferior da subfração. Mas parece desnecessário - talvez atop()tenha uma string vazia padrão para o segundo parâmetro ou algo assim.
naught101
Parece que a resposta de @SandyMuspratt foi modificada depois que postei minha resposta para refletir um código semelhante ao meu. :-)
Aren Cambre
9

não é muito difícil adicionar grobs ao gtable e fazer um título extravagante dessa forma,

library(ggplot2)
library(grid)
library(gridExtra)
library(magrittr)
library(gtable)

p <- ggplot() + 
  theme(plot.margin = unit(c(0.5, 1, 1, 1), "cm"))

lg <- list(textGrob("Rainfall", x=0, hjust=0, 
                    gp = gpar(fontsize=24, fontfamily="Skia", face=2, col="turquoise4")),
               textGrob("location", x=0, hjust=0, 
                        gp = gpar(fontsize=14, fontfamily="Zapfino", fontface=3, col="violetred1")),
           pointsGrob(pch=21, gp=gpar(col=NA, cex=0.5,fill="steelblue")))

margin <- unit(0.2, "line")
tg <- arrangeGrob(grobs=lg, layout_matrix=matrix(c(1,2,3,3), ncol=2),
                  widths = unit.c(grobWidth(lg[[1]]), unit(1,"null")),
                  heights = do.call(unit.c, lapply(lg[c(1,2)], grobHeight)) + margin)

grid.newpage()
ggplotGrob(p) %>%
  gtable_add_rows(sum(tg$heights), 0) %>%
  gtable_add_grob(grobs=tg, t = 1, l = 4)  %>%
  grid.draw()

insira a descrição da imagem aqui

baptista
fonte
8

Esta versão usa uma gtablefunção. Permite duas linhas de texto no título. O texto, tamanho, cor e tipo de letra de cada linha podem ser definidos independentemente um do outro. No entanto, a função irá modificar um gráfico com um único painel de gráfico apenas.

Edição secundária: Atualizando para ggplot2 v2.0.0

# The original plot
library(ggplot2)

secu <- seq(1, 16, by = 2)
melt.d <- data.frame(y = secu, x = LETTERS[1:8])

m <- ggplot(melt.d, aes(x = x, y = y)) + 
     geom_bar(fill="darkblue", stat = "identity") + 
     labs(x = "Weather    stations", y = "Accumulated Rainfall [mm]") + 
     theme(axis.text.x = element_text(angle = -45, hjust = 0, vjust = 1))


# The function to set text, size, colour, and face
plot.title = function(plot = NULL, text.1 = NULL, text.2 = NULL, 
   size.1 = 12,  size.2 = 12,
   col.1 = "black", col.2 = "black", 
   face.1 = "plain",  face.2 = "plain") {

library(gtable)
library(grid)

gt = ggplotGrob(plot)

text.grob1 = textGrob(text.1, y = unit(.45, "npc"), 
   gp = gpar(fontsize = size.1, col = col.1, fontface = face.1))
text.grob2 = textGrob(text.2,  y = unit(.65, "npc"), 
   gp = gpar(fontsize = size.2, col = col.2, fontface = face.2))

text = matrix(list(text.grob1, text.grob2), nrow = 2)
text = gtable_matrix(name = "title", grobs = text, 
   widths = unit(1, "null"), 
   heights = unit.c(unit(1.1, "grobheight", text.grob1) + unit(0.5, "lines"), unit(1.1,  "grobheight", text.grob2) + unit(0.5, "lines")))

gt = gtable_add_grob(gt, text, t = 2, l = 4)
gt$heights[2] = sum(text$heights)

class(gt) =  c("Title", class(gt))

gt
}

# A print method for the plot
print.Title <- function(x) {
   grid.newpage()   
   grid.draw(x)
}


# Try it out - modify the original plot
p = plot.title(m, "Rainfall", "Location", 
   size.1 = 20, size.2 = 15, 
   col.1 = "red", col.2 = "blue", 
   face.2 = "italic")

p

insira a descrição da imagem aqui

Sandy Muspratt
fonte
3

Você pode usar envolver o gráfico em grid.arrange e passar um título baseado em grade personalizado,

insira a descrição da imagem aqui

library(ggplot2)
library(gridExtra)

p <- ggplot() + 
  theme(plot.margin = unit(c(0.5, 1, 1, 1), "cm"))

tg <- grobTree(textGrob("Rainfall", y=1, vjust=1, gp = gpar(fontsize=25, face=2, col="black")),
               textGrob("location", y=0, vjust=0, gp = gpar(fontsize=12, face=3, col="grey50")),
               cl="titlegrob")
heightDetails.titlegrob <- function(x) do.call(sum,lapply(x$children, grobHeight))

grid.arrange(p, top = tg)
baptista
fonte
As versões atuais do ggplot usam theme and element_text em vez de opts e theme_text respectivamente. Além disso, a abordagem ggplotGrob parece falhar na versão atual do gridExtra (gridExtra_0.8.1) e ggplot2 (ggplot2_0.9.3.1)
russellpierce
2

Você deve ter notado que o código de Sandy não produz um título em negrito para "Rainfall" - a instrução para torná-lo em negrito deve ocorrer na função atop () em vez da função theme ().

ggplot(melt.d, aes(x=x, y=y)) + 
 geom_bar(fill="darkblue", stat = "identity") + 
 labs(x="Weather    stations", y="Accumulated Rainfall [mm]") + 
 ggtitle(expression(atop(bold("Rainfall"), atop(italic("Location"), "")))) +
 theme(axis.text.x = element_text(angle=-45, hjust=0, vjust=1),
 plot.title = element_text(size = 25, colour = "black", vjust = -1))

insira a descrição da imagem aqui

Nathan
fonte