R语言技术问询:基于DataFrame因子列生成指定规则的valuetofind向量
解决DataFrame生成特定规则向量的问题
我来帮你搞定这个问题!你遇到的核心问题是因子转数值时的顺序偏差,还有如何按行判断全1/全-1的情况,一步步来解决:
第一步:修正因子转数值的问题
你的DataFrame列都是因子类型,水平是"-1"和"1",直接用as.numeric()会返回1和2(对应因子水平的默认排序),所以得先把因子转成字符再转成数值,才能得到你要的-1和1:
# 将所有因子列转换为正确的数值型 results_num <- data.frame(lapply(results, function(col) as.numeric(as.character(col))))
第二步:生成valuetofind向量
有两种简洁的方法可以实现你的需求:
方法一:用apply逐行判断
逐行检查每一行的所有值是否全为1或全为-1,对应赋值:
results$valuetofind <- apply(results_num, 1, function(row) { if (all(row == 1)) { 1 } else if (all(row == -1)) { -1 } else { NA } })
方法二:用rowSums简化判断
因为你的DataFrame有4列,全1的行求和结果是4,全-1的行求和结果是-4,用这个特性可以简化代码:
results$valuetofind <- ifelse(rowSums(results_num) == 4, 1, ifelse(rowSums(results_num) == -4, -1, NA))
额外简化:无需单独生成数值型DataFrame
如果你不想单独创建results_num,可以在apply里直接完成类型转换和判断:
results$valuetofind <- apply(results, 1, function(row) { row_num <- as.numeric(as.character(row)) if (all(row_num == 1)) 1 else if (all(row_num == -1)) -1 else NA })
这样就能完美生成你需要的valuetofind向量啦!
内容的提问来源于stack exchange,提问作者henry dupuis
相关产品推荐
相关产品推荐

