根据长度重新编码变量

Question

我有一个大型数据框，其结构如下:

id v1 v2 v3 v4 v5
1  1  1 98  1  1
2  1  1  1  1  1
3  4  1  0 22  1
4  5  1  1  1  1
5  1  1 90  1  1 

我想从v2一直移动到v5，如果变量值的长度大于1个字符，那么它会被重新编码为9，所以结果df将是:

id v1 v2 v3 v4 v5
1  1  1  9  1  1
2  1  1  1  1  1
3  4  1  0  9  1
4  5  1  1  1  1
5  1  1  9  1  1 

• 注意:所有变量都存储为字符串，这就是为什么我希望将长度作为答案的一部分。

Answer 1

如果这是一个大型dataframe，使用data.table库，您可以这样做:

Reprex

• 代码

library(data.table)

cols <- paste0("v", 2:5)
setDT(df)[, (cols) := lapply(.SD, function(x) fifelse(nchar(x) > 1, 9, x)), .SDcols = cols][]

• 输出

#>    id v1 v2 v3 v4 v5
#> 1:  1  1  1  9  1  1
#> 2:  2  1  1  1  1  1
#> 3:  3  4  1  0  9  1
#> 4:  4  5  1  1  1  1
#> 5:  5  1  1  9  1  1

^{由 reprex package 于 2022 年 3 月 14 日创建(v2.0.1)}

---

编辑:

dplyr 解决方案

• 代码

library(dplyr)

df %>% mutate(across(v2:v5, ~ ifelse(nchar(.x) > 1, 9, .x)))

• 输出

#>   id v1 v2 v3 v4 v5
#> 1  1  1  1  9  1  1
#> 2  2  1  1  1  1  1
#> 3  3  4  1  0  9  1
#> 4  4  5  1  1  1  1
#> 5  5  1  1  9  1  1

基础 R 解决方案

• 代码

cols <- paste0("v", 2:5)
df[, cols] <- apply(df[, cols], c(1,2), function(x) ifelse(nchar(x) > 1, 9, x))

• 输出

df
#>   id v1 v2 v3 v4 v5
#> 1  1  1  1  9  1  1
#> 2  2  1  1  1  1  1
#> 3  3  4  1  0  9  1
#> 4  4  5  1  1  1  1
#> 5  5  1  1  9  1  1

^{由 reprex package 于 2022 年 3 月 14 日创建(v2.0.1)}