2017-10-19 105 views
2

我想讓我的情節中的動畫線在顏色高於50%時改變顏色,以顯示結果的差異。ggplot2和gganimate:如何使動畫線根據Y軸值改變顏色

我 「data.thesis」 框架是這樣的:

image1

我假設的結局情節是在這裏。從而使黑線移動到可能的結果的代碼將動畫的情節:

image2

library(ggplot2) 
library(gganimate) 

z <- qnorm(0.33) 
sd <- (50 - 55)/z 
moe <- 1.96*sd 

data.thesis <- data.frame(name = "Candidate", percent.vote = 55) 
data.thesis <- data.thesis %>% 
    mutate(sd = sd, 
     ymin = percent.vote - 1.96*sd, 
     ymax = percent.vote + 1.96*sd) 


set.seed(2016) 
n.outcomes <- 50 

df <- data.frame(simulation = 1:n.outcomes, 
       team = "Yes", 
       vote.share = round(rnorm(50, mean = 55, sd = sd), 3)) 

p <- df %>% 
    ggplot(aes(x = team, y = vote.share)) + 
    geom_errorbar(aes(
    ymin = vote.share, ymax = vote.share, 
    frame = simulation, cumulative = TRUE), 
    color = "grey80", alpha = 1/8) + 
    geom_errorbar(aes(
    ymin = vote.share, ymax = vote.share, frame = simulation), 
    color = "black") + 
    theme(panel.background = element_rect(fill = "#FFFFFF"), 
     text=element_text(size=15, vjust=1, family="Verdana"), 
     axis.title.x=element_blank(), 
     axis.text.x=element_blank(), 
     axis.ticks.x=element_blank()) + 
    labs(title = "", y = "Vote share", x = "") + 
    scale_y_continuous(limits=c(0, 100), 
        breaks=c(0, 10, 20, 30, 40, 50, 60, 70, 80, 90, 100)) + 
    xlab("") + 
    ylab("Percent") + 
    ggtitle("Treatment 2: Hypothetical Outcome Plot") + 
    annotate("segment", x=1.6, xend=1.6, y=51, yend=70, colour="black", size=1, arrow=arrow()) + 
    annotate(geom="text", x=1.6, y=74, label="A Wins", color="black", size=5) + 
    annotate("segment", x=1.6, xend=1.6, y=49, yend=30, colour="black", size=1, arrow=arrow()) + 
    annotate(geom="text", x=1.6, y=26, label="A Loses", color="black", size=5) + 
    annotate("text", x = 1.8, y = 75, label = "") + 
    annotate("segment", x=.55, xend=1.45, y=55, yend=55, colour="red") 

gganimate(p, title_frame = FALSE) 

回答

0

而不是硬編碼的線條顏色,用色彩美學在第二次調用geom_errorbar。具體來說,改變從這個第二geom_errorbar聲明:

geom_errorbar(aes(ymin = vote.share, ymax = vote.share, frame = simulation), 
    color = "black") + 

這樣:

geom_errorbar(aes(ymin = vote.share, ymax = vote.share, frame = simulation, 
    colour=ifelse(vote.share > 50, "Win","Lose")), show.legend=FALSE) + 

這是不特定於gganimate,因爲這是你會怎麼做任何ggplot使線顏色取決於超過某個y軸切割點。

在下面的圖中,我添加了+ scale_colour_manual(values=c(Win="blue", Lose="red")),但您可以將顏色設置爲您希望的任何值。另外,我在y = 55處留下了紅線標註。我不確定你是否想在最後的情節中,但它當然可以很容易地被刪除。

enter image description here

+0

這工作完美。非常感謝! –