首页 > 解决方案 > R在数据框中找到值(因子变量的行)

问题描述

我有一个带有这样的字符串(因子)变量的df

id v1 v2 v3 v4
1  1  1  1  0
1  0  0  0  0
1  1  1  1  0
1  1  1  1  0
1  1  1  1  0

我想要的是创建一个等于 1 或 0(数字)的新列,如果列列表中出现或不显示 1 说 (v2, v3, v4)

id v1 v2 v3 v4 lu
1  1  1  1  0  1
2  0  0  0  0  0
3  1  1  1  0  1
4  0  0  1  0  1
5  1  0  1  0  1

我不想转换为数字(因为我需要它们以供将来分析)所以像

df <- df %>% mutate(lu = select(., v2:v4) %>% rowSums(na.rm = TRUE))
df <- df %>% mutate(lu = if_else(lu_1 > 0, 1, lu))

不可能。实际上要考虑的变量列表相当长(大约 150 个)

谢谢!

标签: rtidyverse

解决方案


嗨,我不知道你是否想要这样的东西。我不擅长 dplyr 所以这是一个基本的 R 版本

虚拟数据:

a <-as.character(sample(0:1,100,replace = T))
b <-as.character(sample(0:1,100,replace = T))
c <-as.character(sample(0:1,100,replace = T))
d <-as.character(sample(0:1,100,replace = T))

df <- as.data.frame(cbind(a,b,c,d), stringsAsFactors = T)

实际代码:

test <- function(x) {
  out <- sum(as.numeric(x))
  ifelse(out>0,1,0)
}

df$lu <- as.factor(apply(df,MARGIN = 1, test))

输出:

id  a   b   c   d   lu
1   0   1   0   0   1
2   0   1   0   0   1
3   0   1   0   0   1
4   0   1   0   1   1
5   0   0   0   0   0

推荐阅读