7

我的问题是：我有一个包含一些因子变量的数据框。我现在想为这个数据框分配一个新向量，它为这些因子变量的每个子集创建一个索引。

   data <-data.frame(fac1=factor(rep(1:2,5)), fac2=sample(letters[1:3],10,rep=T))

给了我类似的东西：

        fac1 fac2
     1     1    a
     2     2    c
     3     1    b
     4     2    a
     5     1    c
     6     2    b
     7     1    a
     8     2    a
     9     1    b
     10    2    c

我想要的是一个组合计数器，它计算每个因素组合的出现。像这样

        fac1 fac2  counter
     1     1    a        1
     2     2    c        1
     3     1    b        1
     4     2    a        1
     5     1    c        1
     6     2    b        1
     7     1    a        2
     8     2    a        2
     9     1    b        2
     10    1    a        3

到目前为止，我考虑过使用 tapply 来获取所有因子组合的计数器，效果很好

counter <-tapply(data$fac1, list(data$fac1,data$fac2), function(x) 1:length(x))

但我不知道如何在不使用低效循环的情况下将计数器列表（例如未列出）分配给数据帧中的组合：）

4

4 回答 4

6

这是该ave()功能的工作：

# Use set.seed for reproducible examples 
#   when random number generation is involved
set.seed(1) 
myDF <- data.frame(fac1 = factor(rep(1:2, 7)), 
                   fac2 = sample(letters[1:3], 14, replace = TRUE), 
                   stringsAsFactors=FALSE)
myDF$counter <- ave(myDF$fac2, myDF$fac1, myDF$fac2, FUN = seq_along)
myDF
#    fac1 fac2 counter
# 1     1    a       1
# 2     2    b       1
# 3     1    b       1
# 4     2    c       1
# 5     1    a       2
# 6     2    c       2
# 7     1    c       1
# 8     2    b       2
# 9     1    b       2
# 10    2    a       1
# 11    1    a       3
# 12    2    a       2
# 13    1    c       2
# 14    2    b       3

注意步骤stringsAsFactors=FALSE中的使用data.frame()。如果你没有那个，你仍然可以得到输出：myDF$counter <- ave(as.character(myDF$fac2), myDF$fac1, myDF$fac2, FUN = seq_along)。

于 2012-10-25T15:53:11.283 回答
4

数据表解决方案

library(data.table)
DT <- data.table(data)
DT[, counter := seq_len(.N), by = list(fac1, fac2)]
于 2012-10-25T22:35:28.510 回答
0

这是避免（显式）循环的基本 R 方式。

data$counter <- with(data, {
    inter <- as.character(interaction(fac1, fac2))
    names(inter) <- seq_along(inter)
    inter.ordered <- inter[order(inter)]
    counter <- with(rle(inter.ordered), unlist(sapply(lengths, sequence)))
    counter[match(names(inter), names(inter.ordered))]
})
于 2012-10-25T15:42:01.940 回答
0

这是一个带有一点循环的变体（我已将您的变量重命名为“x”，因为“数据”被用于其他方式）：

x <-data.frame(fac1=rep(1:2,5), fac2=sample(letters[1:3],10,rep=T))
x$fac3 <- paste( x$fac1, x$fac2, sep="" )
x$ctr <- 1
y <- table( x$fac3 )
for( i in 1 : length( rownames( y ) ) )
  x$ctr[x$fac3 == rownames(y)[i]] <- 1:length( x$ctr[x$fac3 == rownames(y)[i]] )
x <- x[-3]

不知道这是否对大型 data.frame 有效，但它有效！

于 2012-10-25T15:57:21.920 回答