R语言使用for循环实现两个嵌套数组列表的对应元素相减
解决方案
方法1:基础R for循环实现
完全匹配你需要用循环处理的需求,逻辑清晰易调试:
# 初始化结果列表,长度和输入的外层列表一致 result <- vector("list", length(example1)) for (i in seq_along(example1)) { # 取出对应位置的数组(子元素为单元素列表,用[[1]]提取内部数组) arr1 <- example1[[i]][[1]] arr2 <- example2[[i]][[1]] # 获取需要广播的次数:即example1对应数组的第三维长度 rep_n <- dim(arr1)[3] # 对arr2做广播,重排为和arr1完全一致的维度 arr2_broad <- array(rep(arr2, rep_n), dim = dim(arr1)) # 逐元素相减后套回原嵌套结构存入结果 result[[i]] <- list(arr1 - arr2_broad) }
你可以通过dim(result[[1]][[1]])验证维度,返回结果为6 3 10,和原example1的结构、维度完全匹配。
方法2:mapply简化实现
无需显式写循环,代码更简洁,输出结果和循环版完全一致:
result <- mapply(function(x, y) { arr1 <- x[[1]] arr2 <- y[[1]] list(arr1 - array(rep(arr2, dim(arr1)[3]), dim = dim(arr1))) }, example1, example2, SIMPLIFY = FALSE)
其中SIMPLIFY = FALSE保证输出保持列表结构,不会被自动转换为其他数据类型。
内容的提问来源于stack exchange,提问作者Thanatocoenose
相关产品推荐
相关产品推荐

