如何动态访问多维数组的指定维度（理想情况下无需重塑数组）

嘿，这个问题我之前也碰到过！其实在R里我们完全可以通过构造动态索引列表的方式解决，不用动数组的结构，而且不管数组有多少维度都能通用，超级方便。

我给你拆解下具体的实现思路，你一看就懂：

首先，我们得精准定位到B所在的维度位置。这里用which(vals == "B")比你原来用的grep更直接——毕竟我们要的是精确匹配，用grep反而可能因为正则规则搞出意外，which一步就能拿到准确的维度下标。

接下来是关键：构造动态索引列表。在R的数组索引逻辑里，TRUE就代表取这个维度的所有元素，那我们先创建一个和数组维度数一样长的列表，把所有元素都设为TRUE，然后再把目标维度对应的位置改成1。这样就相当于告诉R：除了这个维度取第一个元素，其他维度全部保留。

最后，我们用do.call函数把这个索引列表传递给数组的索引操作[，因为do.call能帮我们把列表里的每个元素拆成[的单独参数，完美适配任意维度的数组。

给你看完整的可复用代码：

# 假设已经有了数组arr和维度名称向量vals
# 1. 定位B所在的维度位置
target_dim <- which(vals == "B")
# 2. 获取数组的总维度数
dim_num <- length(dim(arr))
# 3. 构造索引列表：所有维度默认取全部元素
index_list <- rep(list(TRUE), dim_num)
# 4. 将目标维度的索引设为1（取第一个元素）
index_list[[target_dim]] <- 1
# 5. 执行动态索引
result <- do.call(`[`, c(list(arr), index_list))

我们用你给的三个例子来测试下，完全能对应上你要的效果：

测试示例1 #

vals = c("A","B","C")
arr = array(runif(5^3), dim = c(5,5,5))

target_dim <- which(vals == "B")
dim_num <- length(dim(arr))
index_list <- rep(list(TRUE), dim_num)
index_list[[target_dim]] <- 1
result <- do.call(`[`, c(list(arr), index_list))

# 这个result和你手动写arr[,1,]的结果完全一致

测试示例2 #

vals = c("A","C","B","D")
arr = array(runif(5^4), dim = c(5,5,5,5))

target_dim <- which(vals == "B")
dim_num <- length(dim(arr))
index_list <- rep(list(TRUE), dim_num)
index_list[[target_dim]] <- 1
result <- do.call(`[`, c(list(arr), index_list))

# 对应你手动写的arr[,,1,]

测试示例3 #

vals = c("B","A","C","D","E")
arr = array(runif(5^5), dim = c(5,5,5,5,5))

target_dim <- which(vals == "B")
dim_num <- length(dim(arr))
index_list <- rep(list(TRUE), dim_num)
index_list[[target_dim]] <- 1
result <- do.call(`[`, c(list(arr), index_list))

# 对应你手动写的arr[1,,,,]

原理其实很简单：我们把每个维度的索引逻辑封装成了动态生成的列表，再通过do.call把列表转换成[函数需要的参数格式，这样不管数组是3维还是10维，代码都能自动适配，再也不用手动写那些冗长的固定索引了。

备注：内容来源于stack exchange，提问作者mri