用分组向量划分向量

Question

我有两个向量，我想将它们合并到一个数据框中。其中一个向量 values 需要分成两列。第二个向量 nc 告知每个观察值的数量。如果 nc 为 1，则 values 中仅给出一个值（进入 val1）并且 999 将写入第二列（val2 ).

划分向量 value 并填充 df 的两列的 r-ish 方法是什么？我怀疑我错过了一些非常明显的东西，但目前无法继续......非常感谢！

set.seed(123)
nc <- sample(1:2, 10, replace = TRUE)
value <- sample(1:6, sum(nc), replace = TRUE)



# result by hand
df <- data.frame(nc = nc, 
               val1 = c(6, 3, 4, 1, 2, 2, 6, 5, 6, 5), 
               val2 = c(999, 5, 999, 6, 1, 999, 6, 4, 4, 999))

Answer 1

我想这就是您要找的。我不确定这是最快的方法，但它应该可以解决问题。

count <- 0
for (i in 1:length(nc)) {
    count <- count + nc[i]
    if(nc[i]==1) {
        df$val1[i] <- value[count]
        df$val2[i] <- 999
    } else {
        df$val1[i] <- value[count-1]
        df$val2[i] <- value[count]
    }
}

Answer 2

这是一种基于this answer的方法：

set.seed(123)
nc <- sample(1:2, 10, replace = TRUE)
value <- sample(1:6, sum(nc), replace = TRUE)

splitUsing <- function(x, pos) {
    unname(split(x, cumsum(seq_along(x) %in% cumsum(replace(pos, 1, pos[1] + 1)))))
}

combineValues <- function(vals, nums) {
    mydf <- data.frame(cbind(nums, do.call(rbind, splitUsing(vals, nums))))
    mydf$V3[mydf$nums == 1] <- 999
    return(mydf)
}

df <- combineValues(value, nc)

用分组向量划分向量

Divide vector with grouping vector

r

vector

dataframe