1

我是编程初学者。因此，这个问题可能很容易解决。我有三个相同维度的矩阵，例如：

A = matrix(1:16,4,4)
B = matrix(rnorm(16,5,1),4,4)
C = matrix(rnorm(16,9,1),4,4)

我想生成一个新矩阵 (D)，其中包含 A 值低于 8 时位置处的所有 B 值。否则，当 A 值等于或大于 8 时，新矩阵 D 应包含值矩阵 C。已经使用“ifelse”功能解决了这个问题：

D = ifelse(A<8,B,C)

但是，这非常缓慢。有没有更快的方法产生这个矩阵 D？提前谢谢了！

4

2 回答 2

1

尝试这个：

D <- (A < 8) * B + (A >= 8) * C

它有点快：

A = matrix(sample(16,1e4,TRUE),100,100)
B = matrix(rnorm(1e4,5,1),100,100)
C = matrix(rnorm(1e4,9,1),100,100)

require(microbenchmark)

microbenchmark(D1 <- (A < 8) * B + (A >= 8) * C, D2 <- ifelse(A<8,B,C))

Unit: microseconds
                             expr      min        lq    median       uq      max neval
 D1 <- (A < 8) * B + (A >= 8) * C  499.102  528.4075  542.2415  554.983  674.206   100
        D2 <- ifelse(A < 8, B, C) 4015.024 4062.5310 4079.4590 4173.564 5512.694   100

identical(D1,D2)

[1] TRUE

编辑：它可以变得更快：

D <- {A < 8} * {B - C} + C

注意花括号而不是括号和单个比较​​。基准测试：

microbenchmark(D1 <- {A < 8} * {B - C} + C, D2 <- ifelse(A<8,B,C))

Unit: microseconds
                                    expr      min       lq    median       uq      max neval
 D1 <- {     A < 8 } * {     B - C } + C  289.050  300.881  310.7105  333.645  496.189   100
               D2 <- ifelse(A < 8, B, C) 4027.037 4057.980 4069.8110 4148.627 5372.173   100

sum(abs(D1-D2))
#[1] 8.304468e-14

但是考虑到C.

于 2013-09-25T12:00:29.763 回答
1

以下是针对大型数组的几种方法：

A <- matrix(1:16,10000,10000)
B <- matrix(rnorm(10000^2),10000,10000)
C <- matrix(rnorm(10000^2),10000,10000)

> invisible(gc())
> system.time(D<-ifelse(A<8,B,C))
   user  system elapsed 
 15.588   6.608  22.237 
> invisible(gc())
> system.time(D<- (A<8)*B+(A>=8)*C)
   user  system elapsed 
  3.104   3.152   6.267 
> invisible(gc())
> system.time({D<-B; w<-which(A>=8); D[w]<-C[w]})
   user  system elapsed 
  2.872   1.416   4.296 
> invisible(gc())
> system.time({D<-B; w<-(A>=8); D[w]<-C[w]})
   user  system elapsed 
  4.200   1.788   5.998 
> invisible(gc())
> system.time(D<- {A<8}*{B-C}+C)
   user  system elapsed 
  2.012   1.996   4.018 
>

所以，至少在我的机器上，最快的精确方法是{D<-B; w<-which(A>=8); D[w]<-C[w]}. Ferdinand.kraft 提出的方法D<- {A<8}*{B-C}+C稍微快一些，但牺牲了一些精度。

于 2013-09-25T13:09:16.433 回答