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

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倍！