DataFrame迭代中逻辑运算返回值异常问题求解

问题场景 #

需要实现的逻辑：对两个DataFrame的对应行元素构成的区间（即两元素的最小值到最大值范围），若区间包含0则标记为Not Significant，否则标记为Significant。原代码在两元素均为负数时错误返回Not Significant，同时存在循环逻辑问题。

原代码及错误输出 #

原代码 #

a<-c(-1,1,-1,-2,1,0,0,-1)
b<-c(1,1,1,-1,1,1,0,0)
a<-as.data.frame(a)
b<-as.data.frame(b)

ab<-matrix(data=NA)

for (i in 1:nrow(a))
    {
      for (n in 1:nrow(b))
      {
        if (a[i,] <= 0 && b[n,] >= 0) {ab[i]<-c('Not Significant')}
        else {ab[i] <-c('Significant')}
      }
        
    }
c<-cbind(a,b,ab)
c

错误输出 #

错误原因 #

1. 逻辑判断不完整：原条件仅覆盖了a<=0 且 b>=0的情况，但区间包含0的场景还包括a>=0 且 b<=0，以及任意元素为0的情况（0本身属于区间）。
2. 循环逻辑错误：使用双重循环遍历所有行组合，导致每行结果被最后一次循环覆盖，实际应该对对应行的元素进行比较。

修正方案 #

方案1：修正循环与判断逻辑 #

直接计算每行两元素的最小、最大值，判断区间是否包含0：

a <- c(-1,1,-1,-2,1,0,0,-1)
b <- c(1,1,1,-1,1,1,0,0)
a <- as.data.frame(a)
b <- as.data.frame(b)

# 初始化结果向量
ab <- character(nrow(a))

for (i in 1:nrow(a)) {
  val_min <- min(a[i,], b[i,])
  val_max <- max(a[i,], b[i,])
  # 区间包含0的条件：最小值<=0 且 最大值>=0
  ab[i] <- ifelse(val_min <= 0 && val_max >= 0, 'Not Significant', 'Significant')
}

c <- cbind(a, b, ab)
print(c)

方案2：向量化操作（更高效） #

利用R的向量化函数避免循环，提升代码效率：

a <- c(-1,1,-1,-2,1,0,0,-1)
b <- c(1,1,1,-1,1,1,0,0)
a <- as.data.frame(a)
b <- as.data.frame(b)

# 向量化判断区间是否包含0
ab <- ifelse((pmin(a$a, b$b) <= 0) & (pmax(a$a, b$b) >= 0), 
             'Not Significant', 
             'Significant')

c <- cbind(a, b, ab)
print(c)

修正后输出 #

内容的提问来源于stack exchange，提问作者cavalcantelucas