具有（i，j）值且无for或while循环的矩阵中的运算

Question

我需要在R中编写一个函数，该函数接收整数n> 1作为输入，并生成一个输出矩阵P，其中P_ {i，j} = min（i，j）for（i，j）= 1,。 ...，N。 此函数不得具有for和while循环。

到目前为止，我已经尝试使用以下代码。

mat <- function(n){
  m <- matrix(0,nrow = n,ncol = n)
  if(row(m) >= col(m)){
    col(m)
  }
  else{
    row(m)
  }
}

我知道使用if条件，row（m）和col（m）我应该能够查看矩阵，但是，我不知道如何设置该条件，我可以拥有row（m）的最小值）和col（m）在（i，j）位置。 我知道我无法满足上述条件，但是到目前为止，我距离最近的条件是最接近的。

下面是一个示例。 如果n = 3，则结果应为：

     [,1] [,2] [,3]
[1,]    1    1    1
[2,]    1    2    2
[3,]    1    2    3

Answer 1

尝试pmin ， row和col

f1 <- function(n = 3) {
  mat <- matrix(nrow = n, ncol = n)
  pmin(row(mat), col(mat))
}


f1()
#     [,1] [,2] [,3]
#[1,]    1    1    1
#[2,]    1    2    2
#[3,]    1    2    3

---

或者使用更有效的outer和pmin

f2 <- function(n = 3) {
  idx <- sequence(n)
  outer(idx, idx, pmin)
}

基准

library(microbenchmark)
n <- 10000
b <- microbenchmark(
  f1 = f1(n),
  f2 = f2(n),
  times = 10
)

library(ggplot2)
autoplot(b)

b
#Unit: seconds
# expr      min       lq     mean   median       uq      max neval cld
#   f1 5.554471 5.908210 5.924173 5.950610 5.996274 6.058502    10   b
#   f2 1.272793 1.298099 1.354428 1.309208 1.464950 1.495362    10  a