創建中的R

Question

塊對角數據幀我有一個數據組，看起來像這樣：

Person Team 
114 1 
115 1 
116 1 
117 1 
121 1 
122 1 
123 1 
214 2 
215 2 
216 2 
217 2 
221 2 
222 2 
223 2 

「團隊」的範圍從1至33，和團隊在大小方面有所不同（即，可以存在5 ，6或7名成員，視團隊而定）。我需要創建一個如下所示的數據集：

1 1 1 1 1 1 1 0 0 0 0 0 0 0 
1 1 1 1 1 1 1 0 0 0 0 0 0 0 
1 1 1 1 1 1 1 0 0 0 0 0 0 0 
1 1 1 1 1 1 1 0 0 0 0 0 0 0 
1 1 1 1 1 1 1 0 0 0 0 0 0 0 
1 1 1 1 1 1 1 0 0 0 0 0 0 0 
1 1 1 1 1 1 1 0 0 0 0 0 0 0 
0 0 0 0 0 0 0 1 1 1 1 1 1 1 
0 0 0 0 0 0 0 1 1 1 1 1 1 1 
0 0 0 0 0 0 0 1 1 1 1 1 1 1 
0 0 0 0 0 0 0 1 1 1 1 1 1 1 
0 0 0 0 0 0 0 1 1 1 1 1 1 1 
0 0 0 0 0 0 0 1 1 1 1 1 1 1 
0 0 0 0 0 0 0 1 1 1 1 1 1 1 

單個塊的大小由團隊中的人數給出。我如何在R中做到這一點？

Answer 1

另一個想法，不過，我想這是不是RStudent的低效率/優雅：

DF = data.frame(Person = sample(100, 21), Team = rep(1:5, c(3,6,4,5,3))) 
DF 

lengths = tapply(DF$Person, DF$Team, length) 
mat = matrix(0, sum(lengths), sum(lengths)) 
mat[do.call(rbind, 
      mapply(function(a, b) arrayInd(seq_len(a^2), c(a, a)) + b, 
        lengths, cumsum(c(0, lengths[-length(lengths)])), 
        SIMPLIFY = F))] = 1 
mat 

Answer 2

您可以使用包Matrix中的bdiag。例如：

> bdiag(matrix(1,ncol=7,nrow=7),matrix(1,ncol=7,nrow=7))