如何高效获取R语言数据框y中包含至少一个负值的行号？

当然可以！你的嵌套循环确实能达成目标，但R的核心优势之一就是向量化操作，用它来替代手动循环不仅代码更简洁，运行效率也会高很多，尤其是处理大数据框的时候。下面给你两种更高效的实现方式：

方法1：用rowSums（推荐，速度最快） #

这是最简洁且性能最优的写法，利用了R的向量化计算特性：

# 获取至少包含一个负值的行号
negative_row_indices <- which(rowSums(y < 0) > 0)

原理说明：

1. y < 0会生成一个和原数据框维度完全一致的逻辑矩阵，每个元素对应原数据框的数值是否小于0（TRUE表示是负值，FALSE表示不是）。
2. rowSums()会对每行的逻辑值求和（TRUE等价于1，FALSE等价于0），求和结果大于0就说明该行至少有一个负值。
3. which()函数用来提取这些满足条件的行的索引（也就是你要的行号）。

方法2：用apply函数 #

如果你更习惯用apply系列函数，也可以这样写：

# 获取至少包含一个负值的行号
negative_row_indices <- which(apply(y < 0, 1, any))

原理说明：

1. 同样先通过y < 0生成逻辑矩阵。
2. apply(..., 1, any)表示对逻辑矩阵的每一行（参数1代表行方向）应用any()函数，只要该行存在至少一个TRUE，就返回TRUE。
3. 最后用which()提取对应的行号。

效率对比 #

• rowSums是底层优化过的向量化操作，运行速度远快于手动循环和apply（apply本质上是循环的封装，只是代码更简洁）。
• 你的原循环会动态扩展temp向量，这种操作在数据量大的时候会非常慢，而且最后还要处理NA值，上面的两种方法直接得到干净的行号列表，无需额外处理。

测试示例 #

用一个小数据框验证一下：

# 构造示例数据框
y <- data.frame(col1 = c(2, -1, 5), col2 = c(3, 4, -6), col3 = c(-7, 8, 9))

# 用rowSums方法
which(rowSums(y < 0) > 0) 
# 输出：1 2 3（因为每行都有至少一个负值）

# 用apply方法
which(apply(y < 0, 1, any))
# 同样输出：1 2 3

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