索引 R 中 4D 数组中的元素子集

Question

我是 R 的新手。我正在使用 cglasso package in R for one of my projects. The coef.glasso 函数 returns 一个维度为 13470 * 5 * 10 * 10 的 4D 数组（对于我的数据集）。这个 4D 数组包含维度为 13470 * 5 的二维系数矩阵，对应于 10 * 10 超参数的网格。我想迭代 10 * 10 超参数的网格并提取与每个超参数对应的 13470 * 5 2D 系数矩阵。 在一行中我的问题是如何从 13470 * 5 * 10 *10 的 4D 矩阵中提取 13470 * 5 的 2D 矩阵？

Answer 1

假设我们有以下数组（替换您的实际尺寸）：

d <- 5:2
arr <- array(seq_len(prod(d)), dim = d)

我们可以将数组拆分为一维和二维切片列表。

list <- split(arr, slice.index(arr, seq_along(d)[-c(1:2)]))

然后，我们可以重新设置维度，我们将在列表中得到第 1 维和第 2 维的二维矩阵。

list <- lapply(list, function(x) {
  dim(x) <- d[1:2]
  x
})

索引 R 中 4D 数组中的元素子集

indexing a subset of elements from a 4D array in R

regression

r