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

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