我有数组arr可以有任意数量的维度,比方说
arr <- array(NA, c(2,3,5))
我需要一个函数getArrValues,它将一个向量作为输入,并使用该向量从数组中提取值:
getArrValues(arr, c(1,2,2)) # = arr[1,2,2]
getArrValues(arr, c(1,2)) # = arr[1,2,]
getArrValues(arr, c(1)) # = arr[1,,]
您有什么想法可以实现这一目标吗?维度始终存在单个索引,索引按数组维度的顺序提供为向量,数组可以具有任何数量的维度,这些维度始终大于或等于输入向量的长度.
最佳答案 在语言上使用计算是非常简单的.
arr <- array(NA, c(2,3,5))
getArrValues <- function(x, i) {
stopifnot(is.array(x), is.numeric(i))
if (!is.integer(i)) i = as.integer(i)
li = as.list(i)[seq_along(dim(x))]
li = lapply(li, function(x) if (!is.null(x)) x else substitute())
eval(
as.call(c(list(as.name("["), as.name("x")), li))
)
}
getArrValues(arr, c(1,2,2)) # = arr[1,2,2]
getArrValues(arr, c(1,2)) # = arr[1,2,]
getArrValues(arr, c(1)) # = arr[1,,]
只需删除eval(调用以调查确切的未评估电话.