r - 使用ddply时保留有序因子

我经常使用ddply。我偶尔使用有序因素。在包含有序因子的数据帧上调用ddply会删除重组数据帧中的所有排序。

我为ddply编写了以下包装程序，该包装程序记录了级别排序，然后将其重新应用于最初订购的任何列:

dat <- data.frame(a=runif(10),b=factor(letters[10:1],
                                 levels=letters[10:1],ordered=TRUE),
                  c = rep(letters[1:2],times=5),
                  d = factor(rep(c('lev1','lev2'),times=5),ordered=TRUE))

#Drops ordering on b and d
dat1 <- ddply(dat,.(c),transform,log_a = log(a))


ddplyKeepOrder <- function(dat,...){
    orderedCols <- colnames(dat)[sapply(dat,is.ordered)]
    levs <- lapply(dat[,orderedCols,drop=FALSE],levels)
    result <- ddply(.data = dat,...)

    ind <- match(orderedCols,colnames(result))
    levs <- levs[!is.na(ind)]
    orderedCols <- orderedCols[!is.na(ind)]
    ind <- ind[!is.na(ind)]
    if (length(ind) > 0){
        for (i in 1:length(ind)){
            result[,orderedCols[i]] <- factor(result[,orderedCols[i]],
                                          levels=levs[[i]],ordered=TRUE)
        }
    }
    return(droplevels(result))
}

#Preserves ordering on b and d
dat2 <- ddplyKeepOrder(dat,.variables = .(c),.fun = transform,log_a = log(a))

我没有彻底检查此功能，因此在某些情况下它可能无法处理。有没有更好/更完整的方法来解决这个问题？我想如果想一想，我可能可以删除for循环。

特别是，我在ddply调用之后进行的检查以查看是否仍然存在原始的有序因素似乎很丑陋，但是我希望该函数能够处理ddply更改存在哪些列的情况，可能会删除有序因素。

有什么想法吗？

最佳答案

我将以下代码用于这些类型的问题(“ddply”而不是“ordered factor”)，它似乎可以毫无问题地处理您的特定示例(除了不同的行名)。

> dat2 <- do.call(rbind, lapply(split(dat, dat$c), transform, log_a=log(a)))
> str(dat2)
'data.frame':   10 obs. of  5 variables:
 $ a    : num  0.216 0.607 0.197 0.171 0.797 ...
 $ b    : Ord.factor w/ 10 levels "j"<"i"<"h"<"g"<..: 1 3 5 7 9 2 4 6 8 10
 $ c    : Factor w/ 2 levels "a","b": 1 1 1 1 1 2 2 2 2 2
 $ d    : Ord.factor w/ 2 levels "lev1"<"lev2": 1 1 1 1 1 2 2 2 2 2
 $ log_a: num  -1.532 -0.499 -1.625 -1.767 -0.227 ...

关于r - 使用ddply时保留有序因子，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/6670964/