R - 根据条件处理数据帧行

Question

我试图了解如何根据条件处理数据帧的行。 拥有这样的数据框架

> d<-data.frame(x=c(0,1,2,3), y=c(1,1,1,0))
> d
  x y
1 0 1
2 1 1
3 2 1
4 3 0

如何为包含零值的所有行添加+1？ （请注意，可以在任何列中找到零），因此结果将如下所示：

  x y
1 1 2
2 1 1
3 2 1
4 4 1

以下代码似乎可以完成部分工作，但只是打印采取操作的行，采取的次数（2）...

> for(i in 1:nrow(d)){
+     d[d[i,]==0,]<-d[i,]+1
+ }
> d
  x y
1 1 2
2 4 1
3 1 2
4 4 1

我确定有一个简单的解决方案，也许是一个应用函数？，但我没有到达那里。

谢谢。

Answer 1

一些可能性：

# 1
idx <- which(d == 0, arr.ind = TRUE)[, 1]
d[idx, ] <- d[idx, ] + 1
# 2
t(apply(d, 1, function(x) x + any(x == 0)))
# 3
d + apply(d == 0, 1, max)

的用法which为载体，例如which(1:3 > 2)是相当普遍的，而它是为基质中使用以下：通过指定arr.ind = TRUE我们得到的是数组索引，即每0坐标：

which(d == 0, arr.ind = TRUE)
     row col
[1,]   1   1
[2,]   4   2

由于我们只对出现零的行感兴趣，因此我取第一列which(d == 0, arr.ind = TRUE)并将d[idx, ] <- d[idx, ] + 1加到这些行中的所有元素中d[idx, ] <- d[idx, ] + 1 。

关于第二种方法， apply(d, 1, function(x) x)将简单地逐行进行并返回相同的行而不进行任何修改。 通过any(x == 0)我们检查特定行中是否有任何零并获得TRUE或FALSE 。 但是，通过写入x + any(x == 0)我们可以根据需要将TRUE或FALSE分别转换为1或0。

现在是第三种方法。 d == 0是一个逻辑矩阵，我们使用apply来遍历它的行。 然后，当将max应用于特定行时，我们再次将TRUE ， FALSE转换为1,0并找到最大元素。 当且仅当该行中有任何零时，此元素为1。 因此， apply(d == 0, 1, max)返回0和1的向量。 最后一点是当我们写A + b ，其中A是矩阵而b是矢量，加法是逐列的。 这样，通过写d + apply(d == 0, 1, max)我们根据需要在d每一列添加apply(d == 0, 1, max) 。