结合mutate（cross）和case_when来根据条件用0填充多列

Question

在dplyr工作流中，当newvar==0时，我尝试在dataframe的每一列中的newvar列后粘贴一个0，否则不做任何操作。我修改了iris数据集：

library(dplyr)
n <- 150 # sample size

iris1 <- iris %>% 
    mutate(id = row_number(), .before = Sepal.Length) %>% 
    mutate(newvar = sample(c(0,1), replace=TRUE, size=n), .before = Sepal.Length ) %>% 
    mutate(across(.[,3:ncol(.)], ~ case_when(newvar==0 ~ 0)))

null

Answer 1

.[,3:ncol(.)]是列的值，而不是实际的列号。使用3:ncol(.)应该可以很好地工作。

通常，也最好避免按位置引用列，而是使用它们的名称。您可以在一个mutate调用中完成此操作。

library(dplyr)

n <- 150

iris %>% 
  mutate(id = row_number(), 
        newvar = sample(c(0,1), replace=TRUE, size=n), 
        across(Sepal.Length:Petal.Width, ~ case_when(newvar==0 ~ 0, 
                                                     newvar == 1 ~ .)))