索引来自 R

问题描述

我刚开始使用 R 。我在我的一个项目中使用 R 中的 cglasso 包。 coef.glasso 函数返回维度为 13470 * 5 * 10 * 10（对于我的数据集）的 4D 数组。这个 4D 数组包含维度为 13470 * 5 的 2D 系数矩阵，对应于 10 * 10 个超参数的网格。我想遍历 10 * 10 个超参数的网格并提取与每个超参数对应的 13470 * 5 2D 系数矩阵。 在一行中我的问题是如何从维度为 13470 * 5 * 10 *10 的 4D 矩阵中提取维度为 13470 * 5 的二维矩阵？

解决方法

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

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

我们可以将数组拆分为第 1 维和第 2 维的切片列表。

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
})