R中列表上的augment和ggplot的问题

时间:2016-10-10 08:35:01

标签: r list ggplot2 spline broom

我的数据集是:

mydata <- data.frame(
  x1= as.factor(1:3), 
  x2= as.factor (4:6), 
  x3= as.factor(7:9), 
  x4= as.factor (2:7), 
  x5= as.factor(1:6), 
  x6= as.numeric(1:18),
  x7= as.numeric(18:35)
  )

我想在x6和x7上做一个样条曲线,但在我将x1,x2,x3,x4,x5分组之前,我这样做:

mydata1 <- mydata%>%
    nest(-(x1:x5))%>%
    mutate(fit = map(data, ~spline(.$x6,.$x7)))

现在我想使用broom :: augment,但spline是一个列表。另一个问题是我想做一个情节(用ggplot2):

ggplot(data= mydata%>%, aes(x = x6, y = x7))+
    geom_line(aes(color = x1, linetype = x2))+
    geom_line(data= mydata1$fit, aes(x= .$x, y= .$y,color = x1,linetype = x2))+
    facet_grid(x3~x4, scales = "free")

但是给我一个错误:“错误:ggplot2不知道如何处理类列表的数据”,因为ggplot想要一个数据帧而不是列表。有没有一种方法可以适合作为数据帧或在列表中使用扩充? 如果我可以使用扩充,我可以这样做:

mydata1 <- mydata%>%
    nest(-(x1:x5))%>%
    mutate(fit = map(data, ~spline(.$x6,.$x7)),
           result = map(fit, augment))%>%
    unnest(result)

所以我在数据框中安装了。我可以使用ggplot。

1 个答案:

答案 0 :(得分:0)

我认为这是你正在寻找的东西:

library(ggplot2)
library(tidyverse)

data_frame(
  x1=as.factor(1:3), 
  x2=as.factor (4:6), 
  x3=as.factor(7:9), 
  x4=as.factor (2:7), 
  x5=as.factor(1:6), 
  x6=as.numeric(1:18),
  x7=as.numeric(18:35)
) -> mydata

mydata %>%
  nest(-(x1:x5))%>%
  mutate(fit=map(data, ~as_data_frame(spline(.$x6, .$x7)))) -> mydata1

此时,mydata$fit是一列包含xy列的数据框。

您可以为绘图执行此操作:

ggplot() +
  geom_line(data=mydata, 
            aes(x=x6, y=x7, color=x1, linetype=x2))+
  geom_line(data=select(mydata1, -data) %>% unnest(fit), 
            aes(x=x, y=y, color=x1, linetype=x2)) +
  facet_grid(x3~x4, scales="free")