用另一个值将列中的值替换为数据框（对所有值都相同）

Question

我的数据框由 21 列组成，对于这个问题只有一列是相关的：我想将 a 列中的值 2 或 3 或 4 或 5 替换为值 1（在同一列中）。

除了为任何值 2、3、4、5 执行下面的代码之外，我正在寻找更优雅的东西：

  df <- df %>% mutate (a = replace(a, a == 2,1))
  df <- df %>% mutate (a = replace(a, a == 3,1))
  df <- df %>% mutate (a = replace(a, a == 4,1))
  df <- df %>% mutate (a = replace(a, a == 5,1))

所以我只知道条件“或”我需要在代码中创建...任何解决方案？

Answer 1

您可以使用across和多个值替换多个列%in% 。 例如，如果要替换列a 、 b 、 c和d中的值，您可以执行以下操作：

library(dplyr)
df <- df %>% mutate(across(a:d, ~replace(., . %in% 2:5, 1)))
#For dplyr < 1.0.0 use `mutate_at`
#df <- df %>% mutate_at(vars(a:d), ~replace(., . %in% 2:5, 1))

在基数 R 中，您可以使用lapply执行此操作：

cols <- c('a','b','c','d')
df[cols] <- lapply(df[cols], function(x) replace(x, x %in% 2:5, 1))