用第一列中的值替换每行的值

Question

我的问题很简单。我有一个数据框，每一行都有不同的数字，超过 100 列。第一列始终是非零数字。我想要做的是将每行（不包括第一列）中的每个非零数字替换为行中的第一个数字（第一列的值）

我会认为 ifelse 和 for 循环遍历行，但必须有一个更简单的矢量化方式来做到这一点......

Answer 1

另一种方法是使用sapply，它比循环更有效。假设您的数据在数据框中df：

df[,-1] <- sapply(df[,-1], function(x) {ind <- which(x!=0); x[ind] = df[ind,1]; return(x)})

在这里，我们将function 应用于df 除了第一列之外的所有列。在function 中，x 依次是这些列中的每一列：

1. 首先使用which 查找列中为零的行索引。
2. 将x中的这些行设置为df第一列的行中的对应值。
3. 返回列

请注意，函数中的操作都是在列上“矢量化”的。也就是说，没有循环列的行。 sapply 的结果是处理后的列的矩阵，它将替换 df 的所有不是第一列的列。

请参阅 this 以获得对 *apply 系列函数的出色评论。

希望这会有所帮助。

Answer 2

由于你的数据不是很大，我建议你使用一个简单的循环

for (i in 1:nrow(mydata))
{
 for (j in 2:ncol(mydata)
  {

    mydata[i,j]<- ifelse(mydata[i,j]==0 ,0 ,mydata[i,1])
  }
 }

Answer 3

假设您的数据框是dat，我有一个全矢量化解决方案供您使用：

mat <- as.matrix(dat[, -1])
pos <- which(mat != 0)
mat[pos] <- rep(dat[[1]], times = ncol(mat))[pos]
new_dat <- "colnames<-"(cbind.data.frame(dat[1], mat), colnames(dat))

---

示例

set.seed(0)
dat <- "colnames<-"(cbind.data.frame(1:5, matrix(sample(0:1, 25, TRUE), 5)),
                    c("val", letters[1:5]))
#  val a b c d e
#1   1 1 0 0 1 1
#2   2 0 1 0 0 1
#3   3 0 1 0 1 0
#4   4 1 1 1 1 1
#5   5 1 1 0 0 0

我上面的代码给出：

#  val a b c d e
#1   1 1 0 0 1 1
#2   2 0 2 0 0 2
#3   3 0 3 0 3 0
#4   4 4 4 4 4 4
#5   5 5 5 0 0 0

---

您想要一个基准？

set.seed(0)
n <- 2000  ## use a 2000 * 2000 matrix
dat <- "colnames<-"(cbind.data.frame(1:n, matrix(sample(0:1, n * n, TRUE), n)),
                    c("val", paste0("x",1:n)))

## have to test my solution first, as aichao's solution overwrites `dat`

## my solution
system.time({mat <- as.matrix(dat[, -1])
            pos <- which(mat != 0)
            mat[pos] <- rep(dat[[1]], times = ncol(mat))[pos]
            "colnames<-"(cbind.data.frame(dat[1], mat), colnames(dat))})
#   user  system elapsed 
#  0.352   0.056   0.410 

## solution by aichao
system.time(dat[,-1] <- sapply(dat[,-1], function(x) {ind <- which(x!=0); x[ind] = dat[ind,1]; x}))
#   user  system elapsed 
#  7.804   0.108   7.919 

我的解决方案快 20 倍！