r - 如何将数据框的列移动到前几列之后的行中？

Question

我有一个数据框，每一行代表一系列学校

edu <- read.table(header=TRUE, text="Elem Mid High
e1 m1 h1
e2 m2 h2
e1 m2 h2
e3 m1 h1")

我想把它变成一个边缘列表

  s1 s2
1 e1 m1
2 e2 m2
3 e1 m2
4 e3 m1
5 m1 h1
6 m2 h2
7 m2 h2
8 m1 h1

对于有向图（通过 igraph 包）。

这是我的做法：

e2m <- edu[,1:2]
m2h <- edu[,2:3]
colnames(e2m) <- c("s1", "s2")
colnames(m2h) <- c("s1", "s2")
schools <- rbind(e2m,m2e)

“学校”包含我想要的内容，但如果我想添加第四列（例如“Uni”），它是迭代的并且变得很麻烦。什么是矢量化的方式来做到这一点？

Answer 1

这是一个可能的解决方案：

len <- seq_along(edu)
a <- head(len, -1)
b <- tail(len, -1)

data.frame(s1=as.character(unlist(edu[, a])), s2=as.character(unlist(edu[, b])))

Answer 2

直接将 OP 的代码翻译成应用程序。这不是矢量化的：

do.call(rbind, lapply(seq(ncol(edu)-1), FUN=function(x){
  r <- edu[,x:(x+1)]
  colnames(r) <- c('s1', 's2')
  r
}

))

Answer 3

使用@Tyler 的方法：

# assuming a new column added
edu$Uni <- as.factor(c("u1", "u2", "u1", "u1"))  

.

rows  <- nrow(edu)
total <- prod(dim(edu))  # ie: nrow(edu) * ncol(edu)  

X <- as.character(unlist(edu))
data.frame(s1=X[1:(total-rows)],  s2=X[(rows+1):total])

结果：

   s1 s2
1  e1 m1
2  e2 m2
3  e1 m2
4  e3 m1
5  m1 h1
6  m2 h2
7  m2 h2
8  m1 h1
9  h1 u1  <~~~ Added "Uni" column
10 h2 u2  <~~~ Added "Uni" column
11 h2 u1  <~~~ Added "Uni" column
12 h1 u1  <~~~ Added "Uni" column

Answer 4

函数接受的具有矩阵输出的替代方案igraph。

t(
  matrix(
   apply(edu,1,function(x) x[c(1,rep(2:(length(x)-1),each=2),length(x))]),
   nrow=2
        )
 )

结果：

     [,1] [,2]
[1,] "e1" "m1"
[2,] "m1" "h1"
[3,] "e2" "m2"
[4,] "m2" "h2"
[5,] "e1" "m2"
[6,] "m2" "h2"
[7,] "e3" "m1"
[8,] "m1" "h1"

并转换为图表：

> graph.edgelist(result)
IGRAPH DN-- 7 8 -- 
+ attr: name (v/c)