使用for循环模拟R语言is.element函数及错误代码排查
原有代码问题分析
- 匹配逻辑完全错误:原代码的
if (x[i])判断逻辑是判断x的第i个元素是否为非0值,完全没有和目标向量y做匹配校验,和需求的“判断x[i]是否属于y”毫无关联。 - 赋值对象错误:循环中每次直接给整个
answer向量赋值,而非给对应下标的answer[i]赋值,最终返回的向量所有值都是最后一次循环的判断结果,和x的每个元素没有对应关系。
修正后实现代码
核心逻辑为:遍历x的每一个元素,再嵌套遍历y的所有元素做相等判断,只要存在任意一个y的元素和当前x的元素相等,就将对应位置的结果标记为TRUE。
simulate_is_element <- function(x, y) { n_x <- length(x) n_y <- length(y) answer <- logical(n_x) # 初始化和x长度相同的逻辑向量,默认都是FALSE for (i in 1:n_x) { # 遍历y的所有元素判断是否和x[i]相等 for (j in 1:n_y) { if (x[i] == y[j]) { answer[i] <- TRUE break # 匹配到就可以跳出y的循环,不用继续判断 } } } return(answer) } # 测试用例 x <- c(3, 0, -2, 0) y <- c(-1, 0, 1) answer <- simulate_is_element(x, y) answer # 输出结果应为 FALSE TRUE FALSE TRUE,和内置is.element结果一致
内容的提问来源于stack exchange,提问作者Jess A
相关产品推荐
相关产品推荐

