Question

我知道函数relevel将指定的级别设置为第一个。我想知道是否有一个内置函数将指定的级别设置为最后一个。如果没有，编写这样一个函数的有效方法是什么？

Answer 1

没有内置功能。你可以这样做：

lastlevel = function(f, last) {
    if (!is.factor(f)) stop("f must be a factor")
    orig_levels = levels(f)
    if (! last %in% orig_levels) stop("last must be a level of f")
    new_levels = c(setdiff(orig_levels, last), last)
    factor(f, levels = new_levels)
}

x = factor(c("a", "b", "c"))
> lastlevel(x, "a")
[1] a b c
Levels: b c a
> lastlevel(x, "b")
[1] a b c
Levels: a c b
> lastlevel(x, "c")
[1] a b c
Levels: a b c
> lastlevel(x, "d")
Error in lastlevel(x, "d") : last must be a level of f

我觉得有点傻，因为我只是写出来，当我可以对stats:::relevel.factor进行微小的修改。改编自relevel的解决方案如下所示：

lastlevel = function (f, last, ...) {
    if (!is.factor(f)) stop("f must be a factor")
    lev <- levels(f)
    if (length(last) != 1L) 
        stop("'last' must be of length one")
    if (is.character(last)) 
        last <- match(last, lev)
    if (is.na(last)) 
        stop("'last' must be an existing level")
    nlev <- length(lev)
    if (last < 1 || last > nlev) 
        stop(gettextf("last = %d must be in 1L:%d", last, nlev), 
            domain = NA)
    factor(f, levels = lev[c(last, seq_along(lev)[-last])])
}

它会检查更多输入并接受数字（例如，last = 2会将第二级移到最后一级）。

Answer 2

包forcats有一个能够很好地完成这项任务的功能。

f <- gl(2, 1, labels = c("b", "a"))

forcats::fct_relevel(f, "b", after = Inf)

#> [1] b a
#> Levels: a b

将因子的级别设置为最后一个

2 个答案: