Question

我有一个多维数组，并尝试将值分配给特定维度。请参阅下面的代码了解我当前的方法。

# Create a array and fill with NA
set.seed(1)
dim_arr <- seq(2, 10)
arr <- array(NA, dim = dim_arr)

# Add values to the dimension 2 and fill other dimensions
add_dim <- 2

# Generate value
values <- runif(prod(dim_arr[-add_dim]))
# Assign value to position 1 of dimension 2
arr[,1,,,,,,,] <- values

values <- runif(prod(dim_arr[-add_dim]))
# Assign value to position 2 of dimension 2
arr[,2,,,,,,,] <- values

我的问题是维度（dim_arr）和更改维度（add_dim）的数量不固定。当参数dim_arr和add_dim改变时，代码＆＃34; arr [，1 ,,,,,,,]＆＃34;我不得不每次都改变它。

是否有任何通用方法来改进我的代码？

如果我的问题不明确，请告诉我。感谢您的任何建议。

Answer 1

执行此操作的正确方法是致电do.call，指定除所需维度之外的每个维度的完整范围（例如do.call("[<-", list(arr, 1:x, 1:y, 3, values)）。有趣的方法是利用 mysterious missing object 在x[,,,z,,,]中生成空参数。

编辑：最后包含一个更无聊但更简单的方法，因为结果是单个TRUE，因为参数等同于缺少的参数。

EDIT2：简化了Hadley的substitute()方法。

replace_dim <- function(arr, rep.val, dim.to.rep, dim.val) {
  dim.list <- replicate(length(dim(arr)), substitute())
  dim.list[dim.to.rep] <- dim.val
  do.call(`[<-`, c(list(arr), dim.list, list(rep.val)))
}
arr <- array(1:8, dim=rep(2, 3))

replace_dim(arr, 555, 3, 1)  # note this doesn't modify `arr`, so you would have to set `arr` to the return value
# , , 1
#      [,1] [,2]
# [1,]  555  555
# [2,]  555  555
# , , 2
#      [,1] [,2]
# [1,]    5    7
# [2,]    6    8

replace_dim(arr, 555, 2, 1)
# , , 1
#      [,1] [,2]
# [1,]  555    3
# [2,]  555    4
# , , 2
#      [,1] [,2]
# [1,]  555    7
# [2,]  555    8

replace_dim(arr, 555, 1, 1)
# , , 1
#      [,1] [,2]
# [1,]  555  555
# [2,]    2    4
# , , 2
#      [,1] [,2]
# [1,]  555  555
# [2,]    6    8

replace_dim(arr, 555, 1, 2)
# , , 1
#      [,1] [,2]
# [1,]    1    3
# [2,]  555  555
# , , 2
#      [,1] [,2]
# [1,]    5    7
# [2,]  555  555

此外，我们在这里使用了555，但您可以使用任何适合您要插入的尺寸的矢量/对象。

replace_dim_boring <- function(arr, rep.val, dim.to.rep, dim.val) {
  dim.list <- as.list(rep(T, length(dim(arr))))
  dim.list[dim.to.rep] <- dim.val
  do.call(`[<-`, c(list(arr), dim.list, list(rep.val)))
}

将值分配给R中的数组的特定维度

1 个答案: