主页 / coding / 问题 / 77957248
Accepted
Gabriel G.
Asked: 2024-02-08 02:50:21 +0800 CST

当列的特定行包含 R 中的字符串时打印列名

  • 772

我有以下数据集

A    B    C    D
1!   0!   1!   0!
0!   .    1!   1!

我需要打印每个匹配的列名称，如下所示

matches_1     matches_0
A,C           B,D
C,D           A

我在用着：

df$matches_1 <- colSums(apply(df, 1, stringr::str_count, "1!"))

计算 1! 的出现次数 在每一行中，但我不确定如何打印列名而不是总和

2 个回答

  1. Best Answer

    使用==代替stringr，然后提取匹配的colnames，最后paste结果得到所需的输出。

    srch <- c("1!", "0!")

data.frame(
  sapply(srch, \(n) 
    apply(df == n, 1, \(x) paste(colnames(df)[x], collapse=","))), check.names=F)
   1!  0!
1 A,C B,D
2 C,D   A
    • 1

  2. 我们可以编写一个自定义函数：

    my_matches <- function(df, match){
  tmp <- data.frame(which(df == match, arr.ind = TRUE))
  tmp <- split(tmp, tmp$row)
  tmp <- sapply(1:length(tmp), \(x) paste0(names(df)[tmp[[x]][,2]], collapse = ","))
  return(tmp)
}

library(dplyr)
df %>% 
  mutate(matches_1 = my_matches(., "1!"),
         matches_0 = my_matches(., "0!"))

    其产生：

       A  B  C  D matches_1 matches_0
1 1! 0! 1! 0!       A,C       B,D
2 0!  . 1! 1!       C,D         A
    • 0

相关问题