主页 / coding / 问题 / 79594325
Accepted
Shreko
Asked: 2025-04-27 03:36:47 +0800 CST

dplyr：查找列值是否是固定列表中任何项目的子字符串并改变值

  • 772

我从一个简单的例子开始，但我不知道如何在 dplyr mutate 中使用

这有效，我得到了“bcd-234”：

library(tidyverse)
list = c("abc-123", 'bcd-234', 'cde-345', 'bcd-987')
s = 'bcd'
list[str_detect(list, s)][1]

但是当我尝试在 dplyr mutate 中使用它时出现错误：

df <- tibble(
  name = c('aaa', 'bbb', 'ccc', 'ddd', 'abc', 'bcd', 'cde')
)


df |> mutate(
  new_name = if_else(any(str_detect(list, name)), list[str_detect(list, name)][1], name)
  )

我收到一个错误：

! Can't recycle `string` (size 4) to match `pattern` (size 7).

谢谢

3 个回答

  1. Best Answer
    res <- df |>
  rowwise() |>
  mutate(
    new_name = if (any(str_detect(list, name))) list(list[str_detect(list, name)]) else list(name)
  )

    > res
#   name  new_name 
#   <chr> <list>   
# 1 aaa   <chr [1]>
# 2 bbb   <chr [1]>
# 3 ccc   <chr [1]>
# 4 ddd   <chr [1]>
# 5 abc   <chr [1]>
# 6 bcd   <chr [2]>
# 7 cde   <chr [1]>

> as.data.frame(res)
  name         new_name
1  aaa              aaa
2  bbb              bbb
3  ccc              ccc
4  ddd              ddd
5  abc          abc-123
6  bcd bcd-234, bcd-987
7  cde          cde-345

    PS. 避免使用列表作为变量名。

    • 2

  2. 不幸的是，baseR的match()，以及这里更相关的charmatch()，无法处理多个匹配。

    我们可以使用replace++和 ++ on%in%来避免。substrsapplytoStringgrepvidplyr::rowwise

    i = X$name %in% substr(l, 1, 3)
X$new_name = replace(X$name, i, sapply(X$name[i], \(j) toString(grepv(j, l))))

    > X
  name         new_name
1  aaa              aaa
2  bbb              bbb
3  ccc              ccc
4  ddd              ddd
5  abc          abc-123
6  bcd bcd-234, bcd-987
7  cde          cde-345

    注意。这是 2025 年 4 月 11 日发布的How About a Twenty-Sixgrepv()中引入的新功能。它“除了默认设置外，其他功能完全相同”，请参阅新闻。（现在已在 SO 上发布广告。）R 4.5.0grep()value=TRUER

    数据

    OP 的数据命名更恰当。

    > dput(X)
structure(list(name = c("aaa", "bbb", "ccc", "ddd", "abc", "bcd", 
"cde")), class = "data.frame", row.names = c(NA, -7L))
> dput(l)
c("abc-123", "bcd-234", "cde-345", "bcd-987")
    • 1

  3. 在基础 R 中，我们可以使用startsWith，outer将名称与相应的字符串replace匹配；给出结果。diag()

    > res1 <- df |> 
+   transform(
+     new_name=outer(a, name, Vectorize(startsWith)) |> 
+       apply(2, \(i) replace(name, i, toString(a[i]))) |> 
+       diag()
+   ) |> print()
  name         new_name
1  aaa              aaa
2  bbb              bbb
3  ccc              ccc
4  ddd              ddd
5  abc          abc-123
6  bcd bcd-234, bcd-987
7  cde          cde-345

    在哪里

    > str(res1)
'data.frame':   7 obs. of  2 variables:
 $ name    : chr  "aaa" "bbb" "ccc" "ddd" ...
 $ new_name: chr  "aaa" "bbb" "ccc" "ddd" ...

    如果您需要"list"专栏，我们稍微调整一下：

    > res2 <- df |> 
+   base::`$<-`('new_name', outer(a, name, Vectorize(startsWith)) |> 
+       apply(2, \(i) replace(name, i, list(a[i]))) |> 
+       list2DF() |> 
+       as.matrix() |> diag()
+   ) |> print()
  name         new_name
1  aaa              aaa
2  bbb              bbb
3  ccc              ccc
4  ddd              ddd
5  abc          abc-123
6  bcd bcd-234, bcd-987
7  cde          cde-345

    在哪里

    > str(res2)
'data.frame':   7 obs. of  2 variables:
 $ name    : chr  "aaa" "bbb" "ccc" "ddd" ...
 $ new_name:List of 7
  ..$ : chr "aaa"
  ..$ : chr "bbb"
  ..$ : chr "ccc"
  ..$ : chr "ddd"
  ..$ : chr "abc-123"
  ..$ : chr  "bcd-234" "bcd-987"
  ..$ : chr "cde-345"

    数据：

    > dput(df)
structure(list(name = c("aaa", "bbb", "ccc", "ddd", "abc", "bcd", 
"cde")), row.names = c(NA, -7L), class = "data.frame")

a <- c("abc-123", 'bcd-234', 'cde-345', 'bcd-987')
    • 1

相关问题