如何用dplyr筛选出水平数小于n的因子型列?
筛选水平数小于指定值的列(dplyr实现)
没问题,用dplyr确实可以轻松搞定这个需求!针对你提到的Titanic数据集场景,我给你分步拆解解决方案:
1. 先处理数据集(和你提供的代码对齐)
首先把Titanic转换成数据框,并将字符列转为因子:
library(dplyr) titanic_df <- as.data.frame(Titanic) %>% mutate_if(is.character, factor)
2. 筛选水平数小于4的因子列
用select()结合where()函数,搭配nlevels()来判断因子的水平数:
# 筛选水平数 < 4 的因子列 titanic_filtered <- titanic_df %>% select(where(~ is.factor(.) && nlevels(.) < 4))
3. 验证结果
运行str(titanic_filtered)就能看到结果——会保留那3个含2个水平的因子列,自动排除掉有4个水平的列。
补充:如果要扩展到所有列的唯一值筛选
如果你的需求不限于因子列(比如想筛选所有列中唯一值数量小于n的),可以用n_distinct()替代nlevels():
# 筛选所有列中唯一值数量 <4 的列 titanic_filtered_all <- titanic_df %>% select(where(~ n_distinct(.) < 4))
这个逻辑很灵活,你只需要把<4改成你需要的目标数值n就行啦!
内容的提问来源于stack exchange,提问作者Wlademir Ribeiro Prates
相关产品推荐
相关产品推荐

