Início / coding / Perguntas / 77957248
Accepted
Gabriel G.
Asked: 2024-02-08 02:50:21 +0800 CST

imprimir o nome da coluna quando uma linha específica de uma coluna contém uma string em R

  • 772

Eu tenho o seguinte conjunto de dados

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

Preciso imprimir o nome da coluna para cada correspondência da seguinte maneira

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

Estou a usar:

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

para contar ocorrências de 1! em cada linha, mas não sei como imprimir o nome da coluna em vez da soma

2 respostas

  1. Best Answer

    Usando ==em vez de e stringr, em seguida, extraindo a correspondência colnamese, finalmente, pasteo resultado para obter a saída desejada.

    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. Podemos escrever uma função personalizada:

    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!"))

    que produz:

       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

relate perguntas