具有多种线条颜色的facet_grid

时间:2018-05-09 15:08:16

标签: r ggplot2 tidyverse melt

我有以下数据帧,这是由具有不同参数集的ODE模拟得出的,例如

df <- data.frame(t = rep(seq(0,4), 4),
                 x1 = c(1.2*seq(1,5), 1.3*seq(1,5), 1.4*seq(1,5), 1.5*seq(1,5)),
                 x2 = c(0.2*seq(1,5), 0.3*seq(1,5), 0.4*seq(1,5), 0.5*seq(1,5)),
                 a = rep(c(rep(1, 5), rep(2,5)), 2),
                 b = c(rep(1, 10), rep(2,10))
)

我现在想要一个facet_grid,其中x1和x2位于顶部,a和b位于右侧,a和b的值决定了线条的颜色。

我试过

df.1 <- df %>%
  gather(x, xval, -t, -a, -b) %>%
  gather(p, pval, -t, -x, -xval) %>%
  distinct()

df.1$pval <- as.factor(df.1$pval)

ggplot(df.1, aes(t, xval)) +
  geom_line(aes(colour = pval)) +
  facet_grid(p~x)

dm.1 <- melt(df[, c("t", "x1", "x2")], id = 't')
colnames(dm.1) <- c("t", "x", "xval")
dm.2 <- melt(df[, c("t", "a", "b")], id = 't')
colnames(dm.2) <- c("t", "p", "pval")
dm <- merge(dm.1, dm.2)

dm$pval <- as.factor(dm$pval)
ggplot(dm, aes(t, xval)) +
  geom_line(aes(colour = pval)) +
  facet_grid(p~x)

但两者都没有给出理想的结果。任何提示都将不胜感激。

编辑:所需的结果是每个方面中有两条线类似于我的第一个解决方案,但是正确的方法,即直线而不是产生的之字形线。

1 个答案:

答案 0 :(得分:0)

引起之字形图的问题是,px的相同组合有多次重复,但是没有&#39; ta将一个与另一个划分的方式。所以你得到下面的第一个图:

library(tidyverse)

df <- data.frame(t = rep(seq(0,4), 4),
                                 x1 = c(1.2*seq(1,5), 1.3*seq(1,5), 1.4*seq(1,5), 1.5*seq(1,5)),
                                 x2 = c(0.2*seq(1,5), 0.3*seq(1,5), 0.4*seq(1,5), 0.5*seq(1,5)),
                                 a = rep(c(rep(1, 5), rep(2,5)), 2),
                                 b = c(rep(1, 10), rep(2,10))
)

df_long <- df %>%
    gather(key = x, value = xval, x1, x2) %>%
    gather(key = p, value = pval, a, b) %>%
    mutate(pval = as.factor(pval))

df_long %>%
    ggplot(aes(x = t, y = xval)) +
        geom_line(aes(color = pval)) +
        facet_grid(p ~ x)

当我过滤一对特定的xp值时,您可以看到此处的内容。这只是重复地将数据放在t的相同值,而不是知道如何制作不同的行。

df_long %>%
    filter(x == "x1", p == "a") %>%
    head()
#>   t  x xval p pval
#> 1 0 x1  1.2 a    1
#> 2 1 x1  2.4 a    1
#> 3 2 x1  3.6 a    1
#> 4 3 x1  4.8 a    1
#> 5 4 x1  6.0 a    1
#> 6 0 x1  1.3 a    2

相反,在收集之前,您可以为ab的每个组合创建一个ID,并将其用作aes中的分组变量。可能有其他方法可以做到这一点,但一个简单的方法就是interaction(a, b),它会提供类似1.1, 1.2, 2.1, 2.2的ID等。然后在group = id中添加aes制作单独的行。

df_long_id <- df %>%
    mutate(id = interaction(a, b)) %>%
    gather(key = x, value = xval, x1, x2) %>%
    gather(key = p, value = pval, a, b) %>%
    mutate(pval = as.factor(pval))

df_long_id %>%
    ggplot(aes(x = t, y = xval, group = id)) +
    geom_line(aes(color = pval)) +
    facet_grid(p ~ x)

reprex package(v0.2.0)创建于2018-05-09。