如何通过指定矩阵引用并操作R数据框中的特定值

我需要根据一个矩阵的标记，对数据框中的特定值进行替换操作（比如替换为-）。矩阵的行对应数据框里的Name列值，列对应数据框里的Value1到Value4列，矩阵中TRUE的位置就是需要替换的单元格。

最小可复现示例（MRE） #

# 构建原数据框
data <- data.frame(anotherCol  = 1:5,
                   anotherCol2 = 6:10,
                   Name        = c('A', 'B', 'A', 'C', 'C'),
                   Value1      = sample(1:99, 5, TRUE),
                   Value2      = sample(1:99, 5, TRUE),
                   Value3      = sample(1:99, 5, TRUE),
                   Value4      = sample(1:99, 5, TRUE))

# 构建标记矩阵：行是Name值，列是Value列名，TRUE表示需要替换
del <- matrix(sample(c(TRUE,FALSE), 12, TRUE),
              nrow = 3,
              dimnames = list(name  = c('A', 'B', 'C'),
                              value = paste0('Value', 1:4)))

运行后得到的原数据框：

> data
  anotherCol anotherCol2 Name Value1 Value2 Value3 Value4
1          1           6    A     29     53     49     39
2          2           7    B     15     83     54      6
3          3           8    A     46     33     16     59
4          4           9    C     47     99     22     12
5          5          10    C     58     91     24     49

标记矩阵：

> del
    value
name Value1 Value2 Value3 Value4
   A   TRUE  FALSE   TRUE   TRUE
   B  FALSE   TRUE   TRUE  FALSE
   C   TRUE   TRUE  FALSE  FALSE

预期输出 #

> data
  anotherCol anotherCol2 Name Value1 Value2 Value3 Value4
1          1           6    A      -     53      -      -
2          2           7    B     15      -      -      6
3          3           8    A      -     33      -      -
4          4           9    C      -      -     22     12
5          5          10    C      -      -     24     49

解决方案 #

方法1：基础R实现（无需额外包）

# 提取所有以Value开头的列名
value_cols <- grep("^Value", colnames(data), value = TRUE)

# 逐行处理：匹配当前行的Name，找到需要替换的列并修改
for (i in 1:nrow(data)) {
  current_name <- data$Name[i]
  # 筛选出当前Name下需要替换的Value列
  cols_to_replace <- value_cols[del[current_name, ]]
  # 将对应单元格替换为"-"
  data[i, cols_to_replace] <- "-"
}

# 查看处理后的数据框
data

方法2：tidyverse工具包实现（更简洁）

如果习惯用tidyverse生态的工具，这种方法更直观，适合复杂数据处理场景：

library(dplyr)
library(tidyr)

# 先把标记矩阵转换为长格式数据框，方便后续匹配
del_tidy <- as.data.frame(del) %>%
  rownames_to_column("Name") %>%
  pivot_longer(cols = starts_with("Value"), names_to = "Col", values_to = "need_replace") %>%
  filter(need_replace) %>%
  select(Name, Col)

# 原数据框转长格式，匹配需要替换的行，替换后转回宽格式
data_processed <- data %>%
  pivot_longer(cols = starts_with("Value"), names_to = "Col", values_to = "Value") %>%
  left_join(del_tidy, by = c("Name", "Col")) %>%
  # 标记需要替换的单元格改为"-"，其余保留原数值
  mutate(Value = ifelse(!is.na(need_replace), "-", as.character(Value))) %>%
  select(-need_replace) %>%
  pivot_wider(names_from = "Col", values_from = "Value") %>%
  select(colnames(data)) # 保持原数据框的列顺序

# 查看结果
data_processed

两种方法都能得到预期的输出，你可以根据自己的习惯选择合适的方式。

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