R语言如何匹配查找表对嵌套列表径流数据标准化且保留原结构

问题原因 #

原有代码未传入当前处理对象对应的模型编号、子流域编号两个匹配维度，无法从查找表中定位到唯一对应的Q_mean值，因此计算逻辑出错。

实现代码 #

基础R实现（无需额外安装包） #

# 标准化计算函数，传入匹配所需的模型编号、子流域编号
fun_st <- function(df, model_id, subbasin_id, lookup_df = FlowtestDF) {
  # 双条件匹配对应均值
  target_mean <- lookup_df$Q_mean[lookup_df$test_model == model_id & lookup_df$test_subbasin == subbasin_id]
  df$Q_st <- df$Q / target_mean
  return(df)
}

# 两层遍历保留原结构
testresult <- lapply(seq_along(Flowtestlist), function(model_idx) {
  current_sub_list <- Flowtestlist[[model_idx]]
  # 内层遍历保留子流域命名
  lapply(names(current_sub_list), function(sub_name) {
    fun_st(current_sub_list[[sub_name]], 
           model_id = model_idx, 
           subbasin_id = as.integer(sub_name))
  }) |> setNames(names(current_sub_list))
})

purrr 简洁实现 #

library(purrr)
testresult <- imap(Flowtestlist, function(sub_list, model_idx) {
  imap(sub_list, function(df, sub_name) {
    target_mean <- FlowtestDF$Q_mean[FlowtestDF$test_model == as.integer(model_idx) & FlowtestDF$test_subbasin == as.integer(sub_name)]
    df$Q_st <- df$Q / target_mean
    return(df)
  })
})

效果说明 #

• 完全保留原嵌套列表的层级结构、子流域命名规则，和原有列表布局完全一致
• 所有Q_st计算严格匹配对应模型+子流域的Q_mean值，匹配逻辑无歧义

内容的提问来源于stack exchange，提问作者Joanna OK