创建从数据帧

Question

矩阵我有5列c1，c2，c3，c4和c5数据帧df1：

df1 <- data.frame(c1 = c(1, 0, 1), c2 = c(0, 0, 1), c3 = c(2, 1, 0), c4 = c(1, 4, 3), c5 = c(1, 0, 0)) 

    #df1 
    c1 c2 c3 c4 c5 
    1 0 2 1 1 
    0 0 1 4 0 
    1 1 0 3 0 

我的期望的输出是：

1 3 3 4 5 
    3 4 4 4 4 
    1 2 4 4 4 

基本上，我基于列号和它的值创建矩阵（值总是小于或等于5）。例如，df1中的第二行具有1和4属于列c3和c4。因此，所需矩阵中的第二行将是3 4 4 4 4。

我自己试了一下。我的代码涉及几个嵌套的控制流，对于大型数据框不适用。

Answer 1

小apply魔术：

t(apply(df, 1, function(x) rep(row(as.matrix(x)), x))) 
#  [,1] [,2] [,3] [,4] [,5] 
#[1,] 1 3 3 4 5 
#[2,] 3 4 4 4 4 
#[3,] 1 2 4 4 4 

Answer 2

这里有一个稍微不同的选项

t(apply(df1, 1, FUN = function(x) rep(which(x!=0), x[x!=0]))) 
#  [,1] [,2] [,3] [,4] [,5] 
#[1,] 1 3 3 4 5 
#[2,] 3 4 4 4 4 
#[3,] 1 2 4 4 4