我有一个包含因子的数据帧。当我使用子集或其他索引函数创建这个数据帧的子集时，就创建了一个新的数据帧。然而，因子变量保留其所有原始水平，即使它们不存在于新的数据框架中。
这在绘制面图或使用依赖于因子级别的函数时会导致问题。
在新的数据框架中从一个因子中移除级别最简洁的方法是什么?
这里有一个例子:
df <- data.frame(letters=letters[1:5],
numbers=seq(1:5))
levels(df$letters)
## [1] "a" "b" "c" "d" "e"
subdf <- subset(df, numbers <= 3)
## letters numbers
## 1 a 1
## 2 b 2
## 3 c 3
# all levels are still there!
levels(subdf$letters)
## [1] "a" "b" "c" "d" "e"
当前回答
从R版本2.12开始，就有了一个droplevels()函数。
levels(droplevels(subdf$letters))
其他回答
我写了效用函数来做这个。现在我知道了gdata的drop。水平，看起来很相似。他们在这里(从这里):
present_levels <- function(x) intersect(levels(x), x)
trim_levels <- function(...) UseMethod("trim_levels")
trim_levels.factor <- function(x) factor(x, levels=present_levels(x))
trim_levels.data.frame <- function(x) {
for (n in names(x))
if (is.factor(x[,n]))
x[,n] = trim_levels(x[,n])
x
}
下面是另一种方法，我认为它相当于因子(..)方法:
> df <- data.frame(let=letters[1:5], num=1:5)
> subdf <- df[df$num <= 3, ]
> subdf$let <- subdf$let[ , drop=TRUE]
> levels(subdf$let)
[1] "a" "b" "c"
为了完整起见，现在在forcats包http://forcats.tidyverse.org/reference/fct_drop.html中也有fct_drop。
它与液滴处理NA的方式不同:
f <- factor(c("a", "b", NA), exclude = NULL)
droplevels(f)
# [1] a b <NA>
# Levels: a b <NA>
forcats::fct_drop(f)
# [1] a b <NA>
# Levels: a b
如果你不想要这种行为，不要使用因子，而是使用字符向量。我觉得这比事后修补要好得多。在用read加载数据之前，请尝试以下操作。表或read.csv:
options(stringsAsFactors = FALSE)
缺点是你只能按字母排序。(重新排序是你的朋友情节)
从R版本2.12开始，就有了一个droplevels()函数。
levels(droplevels(subdf$letters))
