R语言如何使用不等长向量列表替换表格指定列的正值

我近期提出过相关问题，但此前设置的示例过于简单，本次增加场景复杂度：向量长度不再与表格长度相等。

现有如下表格：

tableA <- structure(c(1L, 0L, 0L, 0L, 4L, 6L, 0L, 6L, 1L, 3L, 0L, 0L, 0L, 0L, 1L), dim = c(3L, 
                                                                   5L), dimnames = structure(list(c("X", "Y", 
                                                                                                    "Z"), c("A", "B", "C","D", "E")), names = c("", "")), class = "table")

    A B C D E
  X 1 0 0 3 0 (two positive numbers)
  Y 0 4 6 0 0 (two positive numbers)
  Z 0 6 1 0 1 (three positive numbers)

同时存在如下向量列表：

listB <- list(
        "X" = c(0, 4),
        "Y" = c(4, 5),
        "Z" = c(7, 1, 0))

各向量长度不相等，但每个向量的元素总数与表格对应行的正值数量一致。需求为：将tableA中B、C、D列所有大于0的值，替换为listB中的对应值，其中listB中值为0的位置不执行替换。

期望输出结果如下：

A B C D E
  X 1 0 0 4 0
  Y 0 4 5 0 0
  Z 0 7 1 0 1 

原有方案提供的代码如下：

cols2replace  = match(c('B', 'C', 'D'), colnames(tableA))
cells2replace = tableA[, cols2replace] > 0
    
tableB = matrix(unlist(listB), nrow = 3, byrow = TRUE)
tableA[, cols2replace][cells2replace] = tableB[, cols2replace][cells2replace]

但在实际数据中，列表内的向量长度不等，因此将listB直接转为固定行数矩阵的代码tableB = matrix(unlist(listB), nrow = 3, byrow = TRUE)无法正常运行。

考虑是否可以先提取tableA中所有正值，用listB中的非零值替换，再将值放回表格。初步编写的代码如下：

# Get all positive values from the table
library(dplyr)
library(tidyr)
library(stringr)
out <- tableA %>% 
   pivot_longer(cols = -rn) %>% 
   filter(str_detect(value, '\b0\b', negate = TRUE)) %>% 
   group_by(rn) %>% 
   summarise(freq = list(value), .groups = 'drop')

但该代码无法直接在table类对象上运行。

直接用基础R逐行匹配替换即可，不需要依赖额外第三方包，逻辑清晰且适配向量长度不等的场景：

# 定义需要执行替换的列范围
replace_cols <- c("B", "C", "D")

for (row_name in names(listB)) {
  # 找到当前行所有值大于0的单元格位置
  positive_idx <- which(tableA[row_name, ] > 0)
  # 取出当前行对应的替换向量
  replace_vals <- listB[[row_name]]
  # 筛选需要替换的位置：替换值非0，且单元格属于指定替换列
  valid_replace <- replace_vals != 0 & colnames(tableA)[positive_idx] %in% replace_cols
  # 执行替换
  tableA[row_name, positive_idx[valid_replace]] <- replace_vals[valid_replace]
}

运行代码后得到的tableA与期望输出完全一致：

A B C D E
  X 1 0 0 4 0
  Y 0 4 5 0 0
  Z 0 7 1 0 1

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