Jak utworzyć wykres liniowy z grupami w Base R bez pętli?

Question

muszę stworzyć prosty liniowy wykres z grup, stosując następujące dane:

test = data.frame(x = rep(1:3, each = 2), 
        group = rep(c("Group 1","Group 2"),3), 
        groupcd= rep(c(1,2),3), 
        y= c(22,8,11,4,7,5) 
       ) 

można łatwo zrobić z ggplot:

library(ggplot2) 
    #GGPLOT 
    qplot(x=x, y=y, 
      data=test, 
      colour=group, 
      main="GGPLOT line plot with groups") + 
     geom_line() 

Mogę też zrobić to z TRELLIS:

library(lattice) 
xyplot(y~x, 
     type="b", 
     group=group, 
     data=test, 
     main="TRELLIS line plot with groups", 
     auto.key =list(
     lines = TRUE) 
     ) 

Jednak jestem trochę niechętny do korzystania z GGPLOT lub TRELLIS już teraz. Chciałbym być w stanie stworzyć ten wykres z podstawą R. jedyny sposób mogę dostać tę działkę do pracy w bazie R jest za pomocą pętli for:

# set up empty plot 
plot(test$y ~test$x, ylab="y", xlab="x", type="n", main="Base R line plot with groups") 
colors<-c("red","blue") 
#plot each group in the for loop 
number_of_groups <- as.numeric(max(unique(test$groupcd))) #calculate number of groups 
for (i in 1:number_of_groups) 
{ 
    temp <- subset(test, groupcd==i) 
    lines(temp$x, temp$y, col=colors[i]) 
    points(temp$x, temp$y, col=colors[i]) 
} 
legend("top", legend=unique(test$group), text.col =colors ) 

Takie podejście wydaje się dość zwinięty. Czy jest łatwiejszy sposób na zrobienie tego w bazie R? Czy istnieje opcja grupowa w podstawowej funkcji fabuły? Dziękuję bardzo.

Answer 1

Co o czymś takim jako baza do pracy z:

test = data.frame(x = rep(1:3, each = 2), 
        group = rep(c("Group 1","Group 2"),3), 
        groupcd= rep(c(1,2),3), 
        y= c(22,8,11,4,7,5) 
       ) 

xvals <- split(test$x,test$group) 
yvals <- split(test$y,test$group) 

plot(1:max(unlist(xvals)),ylim=(c(0,max(unlist(yvals)))),type="n") 
# thanks to @BenBolker for refining this next key line 
mapply(lines,xvals,yvals,col=c("red","blue"),pch=1:2,type="o") 

Wynik:

Answer 2

innego podejścia z wykorzystaniem MATPLOT:

library(reshape) 
test = data.frame(x = rep(1:3, each = 2), 
        group = rep(c("Group 1","Group 2"),3), 
        groupcd= rep(c(1,2),3), 
        y= c(22,8,11,4,7,5) 
       ) 
colors<-c("red","blue") 

#Transform data to wide format 
test_transposed<-reshape(test, 
         idvar='x', 
         drop="group", 
         timevar="groupcd", 
         direction="wide") 
colors<-c("red","blue") 

#drop x column 
test_transposed$x<-NULL 

matplot(test_transposed, 
     type = "b", 
     ylab="y", 
     col=colors, 
     main="MATPLOT with groups", 
     pch = 1:2) 

legend("top", 
     legend=unique(test$group), 
     lty=1:2, 
     col=colors, 
     pch=1:2 ) 

Answer 3

Zastanawiam się również nad tym. Moje dane paneli nie mają pełnego zasięgu na osi X (w latach), więc rozwiązania podobne do macierzy byłyby prawdopodobnie skomplikowane. Zamiast tego poszedłem z ...

mytest = data.frame(
       x = rep(1:3, each = 2), 
       groupcd= rep(c(1,2),3), 
       y= c(22,8,11,4,7,5) 
      ) 
mytest = rbind(mytest,c(2,3,15),c(3,3,17)) 

plottables <- split(mytest,mytest$groupcd) 
plot(y~x,dat=plottables[[1]],type="l",xlim=range(mytest$x),ylim=range(mytest$y)) 
lapply(plottables,function(z)points(y~x,dat=z,type="l")) 

Answer 4

Chciałbym przedłużyć trochę ten post..suppose że instad o wykreślenie linii chcę linii regresji. Rozszerzanie kodu @Frank ..

mytest = data.frame(
       x = rep(1:3, each = 2), 
       groupcd= rep(c(1,2),3), 
       y= c(22,8,11,4,7,5) 
      ) 
mytest = rbind(mytest,c(2,3,15),c(3,3,17)) 

    plottables <- split(mytest,mytest$groupcd) 
plot(y~x,dat=plottables[[1]],type="n",xlim=range(mytest$x),ylim=range(mytest$y)) 
lapply(plottables,function(z)points(z$x, z$y)) 
lapply(plottables,function(z)lines(z$x, predict(lm(z$y ~ z$x), list(z$x)))) 

Teraz załóżmy, że chcę również zmienić kolory dla każdego poziomu grupy. Próbowałem podejście zaproponowane przez @BenBolker

xvals <- tapply(mytest$x, mytest$groupcd, function(x) return(x)) 
    yvals <- tapply(mytest$y, mytest$groupcd, function(x) return(x)) 

    plot(1:max(unlist(xvals)),ylim=(c(0,max(unlist(yvals)))),type="n") 
    mapply(points, xvals, yvals, bg=c(1:3),pch=21,type="p") 
    mapply(lines(xvals, predict(lm(yvals ~ xvals), list(xvals)), col = c(1:3))) 

des ale nie działa.