Question

以下是生成data.frame的代码：

ref_variables=LETTERS[1:10]
row=100
d0=seq(1:100)
for (i in seq_along(ref_variables)){
  dtemp=sample(seq(1:row),row,TRUE)
  d0=data.frame(d0,dtemp)
}
d0[,1]=NULL
names(d0)=ref_variables

我有一个数据集，data.frame或data.table，无论如何。假设我想通过将第2列除以第1列来修改第2列到第4列。当然，我可以这样做一个循环：

columns_name_to_divide=c("B","C","H")
column_divisor="A"
for (i in seq_along(columns_name_to_divide)){        
  ds[columns_name_to_divide[i]] = ds[columns_name_to_divide[i]] / ds[column_divisor]
}

但有没有更优雅的方法呢？

Answer 1

  > d0[2:4] <- d0[,2:4]/d0[,1]

这会将您的原始值替换为将第2,3,4列除以第1列后得到的结果。其余将保持不变。

如果要在第2,3,4列除以第1列之后使用新值在d0中创建3个新列，则不会替换第2,3和4列中的原始值。计算值将分别在第11,12和13栏。

  > dim(d0)
  # [1] 100  10
  > d0[11:13] <- d0[,2:4]/d0[,1]
  > dim(d0)
  # [1] 100  13

要整理新值，您只需将round()函数添加到2位小数位，如下所示：

  > d0[2:4] <- round(d0[,2:4]/d0[,1],2)  # Original values subtituted at 2,3,4

  # OR

  > d0[11:13] <- round(d0[,2:4]/d0[,1],2)  # New columns added, original columns are untouched.

Answer 2

我们可以使用set中的data.table来提高效率，因为多次调用时会避免.[data.table的开销（尽管不是在这种情况下）。

library(data.table)
setDT(d0)
for(j in columns_name_to_divide){
   set(d0, i = NULL, j = j, value = d0[[j]]/d0[[column_divisor]])
}

或使用lapply

setDT(d0)[, (columns_name_to_divide) := lapply(.SD, `/`, 
              d0[[column_divisor]]), .SDcols = columns_name_to_divide]

使用dplyr

的优雅选项

library(dplyr)
library(magrittr)
d0 %<>%
    mutate_each_(funs(./d0[[column_divisor]]), columns_name_to_divide)
head(d0)
#  A         B         C  D  E  F  G        H  I  J
#1 60 0.4000000 1.1500000  6 86 27 19 0.150000 94 97
#2 11 0.6363636 0.3636364 25 52 44 82 8.818182 84 68
#3 80 0.8750000 1.1375000 72 34 56 69 0.125000 34 17
#4 77 0.3116883 1.0259740  9 44 87 61 1.064935 79 40
#5 18 0.3333333 5.0555556 60 69 62 89 2.166667 21 34
#6 42 1.3333333 2.3095238 61 20 87 95 1.428571 78 63

基准

set.seed(42)
d1 <- as.data.frame(matrix(sample(1:9, 1e7*7, replace=TRUE), ncol=7))

d2 <- copy(d1)
d3 <- copy(d1)

system.time({
d2 %<>%
   mutate_each(funs(./d2[["V2"]]), V4:V7)
})
# user  system elapsed 
#   0.52    0.39    0.91 

system.time({
d1[,4:7] <- d1[,4:7]/d1[,2]
})
#   user  system elapsed 
#   1.72    0.72    2.44 


system.time({

setDT(d3)
for(j in 4:7){
   set(d3, i = NULL, j = j, value = d3[[j]]/d3[["V2"]])
}

})
#  user  system elapsed 
#   0.32    0.16    0.47

Answer 3

你可以这样做：

library(data.table)


cols <- names(df)[2:4]
col1 <- names(df)[1]

setDT(df)[, (cols) := lapply (cols, function(x)  get(x) / get(col1) )]


# sample data for reproducible example:
df <- data.frame(V1=rep(10,5),
                 V2=rep(20,5),
                 V3=rep(30,5),
                 V4=rep(40,5),
                 V5=rep(50,5))

通过R中的相同函数修改多列数据集

3 个答案:

基准