AskOverflow.Dev

AskOverflow.Dev Logo AskOverflow.Dev Logo

AskOverflow.Dev Navigation

  • Início
  • system&network
  • Ubuntu
  • Unix
  • DBA
  • Computer
  • Coding
  • LangChain

Mobile menu

Close
  • Início
  • system&network
    • Recentes
    • Highest score
    • tags
  • Ubuntu
    • Recentes
    • Highest score
    • tags
  • Unix
    • Recentes
    • tags
  • DBA
    • Recentes
    • tags
  • Computer
    • Recentes
    • tags
  • Coding
    • Recentes
    • tags
Início / coding / Perguntas / 77780644
Accepted
fe108
fe108
Asked: 2024-01-08 21:35:45 +0800 CST2024-01-08 21:35:45 +0800 CST 2024-01-08 21:35:45 +0800 CST

Verificando se os conjuntos de colunas são iguais, em linha em R, em qualquer ordem

  • 772

Estou trabalhando em R e preferiria uma solução dplyr, se possível.

dados de amostra:

data.frame(
  col1 = c("a", "b", "c", "d"),
  col2 = c("a", "b", "d", "a"),
  col3 = rep("a", 4L),
  col4 = c("a", "b", "d", "a"),
  col5 = c("a", "a", "c", "d"),
  col6 = rep(c("b", "a"), each = 2L)
)
col1 col2 col3 col4 col5 col6
a a a a a b
b b a b a b
c d a d c a
d a a a d a

Pergunta

Gostaria de saber para cada linha se col1, col2 e col3 são iguais a col4, col5 e col6, mas a ordem de col1 - col3 e col4 - col6 deve ser ignorada.

Portanto, para a linha 1, se col1 - col3 contivesse a,a,b respectivamente, e col4 - col6 contivesse b,a,a respectivamente, isso seria considerado uma correspondência.

Resultado desejado

Coloquei uma nota na coluna "avaliação" para ajudar na compreensão

col1 col2 col3 col4 col5 col6 avaliação
a a a a a b FALSO (porque 1-3 não é igual a 4-6)
b b a b a b VERDADEIRO (porque 1-3 é igual a 4-6, se ignorar a ordem)
c d a d c a VERDADEIRO (porque 1-3 é igual a 4-6, se ignorar a ordem)
d a a a d a VERDADEIRO (porque 1-3 é igual a 4-6, se ignorar a ordem)
  • 8 8 respostas
  • 223 Views

8 respostas

  • Voted
  1. s_baldur
    2024-01-08T21:47:29+08:002024-01-08T21:47:29+08:00

    BaseR:

    df$assessment <- apply(df, 1, \(x) identical(table(x[1:3]), table(x[4:6])))
    
    #   col1 col2 col3 col4 col5 col6 assessment
    # 1    a    a    a    a    a    b      FALSE
    # 2    b    b    a    b    a    b       TRUE
    # 3    c    d    a    d    c    a       TRUE
    # 4    d    a    a    a    d    a       TRUE
    

    Dados reproduzíveis:

    df <- data.frame(
      col1 = c("a", "b", "c", "d"), col2 = c("a", "b", "d", "a"),
      col3 = c("a", "a", "a", "a"), col4 = c("a", "b", "d", "a"),
      col5 = c("a", "a", "c", "d"), col6 = c("b", "b", "a", "a")
    )
    

    PS: Por que table() e indentical em vez de sort(), ==, all()? Eu esperaria que fosse melhor dimensionado com o número de colunas (dado o baixo número de valores exclusivos). Exemplo:

    df <- as.data.frame(lapply(1:600, \(x) sample(letters, size = 4000, replace = TRUE)))
    bench::mark(
      apply(df, 1, \(x) identical(table(x[1:300]), table(x[301:600]))),
      apply(df, 1, \(x) all(sort(x[1:300]) == sort(x[301:600])))
    )
    #   expression                                                   min median `itr/sec` mem_alloc 
    #   <bch:expr>                                                 <bch> <bch:>     <dbl> <bch:byt>   
    # 1 apply(df, 1, function(x) identical(table(x[1:300]), table… 1.68s  1.68s     0.594     333MB   
    # 2 apply(df, 1, function(x) all(sort(x[1:300]) == sort(x[301… 9.01s  9.01s     0.111     191MB 
    

    PS 2: Substituir table(x)por collapse::fcount(x, sort = TRUE)dá uma grande aceleração.

    • 11
  2. Best Answer
    Isaac
    2024-01-08T21:42:55+08:002024-01-08T21:42:55+08:00

    Usando dplyr você pode fazer o seguinte:

    df %>%
      rowwise() %>%
      mutate(result = all(sort(c_across(col1:col3)) == sort(c_across(col4:col6))))
    
    # A tibble: 4 × 7
    # Rowwise: 
      col1  col2  col3  col4  col5  col6  result
      <chr> <chr> <chr> <chr> <chr> <chr> <lgl> 
    1 a     a     a     a     a     b     FALSE 
    2 b     b     a     b     a     b     TRUE  
    3 c     d     a     d     c     a     TRUE  
    4 d     a     a     a     d     a     TRUE  
    
    • 8
  3. SamR
    2024-01-08T21:59:34+08:002024-01-08T21:59:34+08:00

    Evite iterar nas linhas

    Geralmente, a iteração nas linhas é lenta e descobri que dplyr::rowwise()as abordagens se tornam muito lentas com mais de alguns milhares de linhas. Tende a ser mais rápido de usar purrr::pmap()e muito mais rápido de iterar nas colunas.

    abordagem base R

    Você poderia pegar a transposição das colunas relevantes e iterar sobre as colunas disso.

    cols <- paste0("col", 1:6)
    
    df$assessment <- df[cols] |>
        t() |>
        data.frame() |>
        sapply(\(x) all(sort(x[1:3]) == sort(x[4:6])))
    
    #   col1 col2 col3 col4 col5 col6 assessment
    # 1    a    a    a    a    a    b      FALSE
    # 2    b    b    a    b    a    b       TRUE
    # 3    c    d    a    d    c    a       TRUE
    # 4    d    a    a    a    d    a       TRUE
    

    tidyverseabordagem: converter de largo para longo

    Alternativamente, se você quiser permanecer no tidyversevocê pode converter de largo para longo:

    df %>%
        mutate(
            assessment = . |>
                mutate(id = row_number()) |>
                tidyr::pivot_longer(
                    -id,
                    names_to = "col",
                    names_transform = readr::parse_number
                ) |>
                group_by(id) |>
                summarise(
                    assessment = all(
                        sort(
                            value[col %in% 1:3]
                        ) ==
                            sort(
                                value[col %in% 4:6]
                            )
                    )
                ) |>
                pull(
                    assessment
                )
        )
    

    Isso é mais detalhado, mas suspeito que será consideravelmente mais rápido com qualquer conjunto de dados de tamanho razoável.

    • 5
  4. jpsmith
    2024-01-08T21:46:34+08:002024-01-08T21:46:34+08:00

    Na base R você poderia usar vapplye sort:

    df$assessment <- vapply(seq_len(nrow(df)), \(x) 
                            all(sort(unlist(df[x,1:3])) == sort(unlist(df[x,4:6]))), logical(1))
    

    Saída:

    #   col1 col2 col3 col4 col5 col6 assessment
    # 1    a    a    a    a    a    b      FALSE
    # 2    b    b    a    b    a    b       TRUE
    # 3    c    d    a    d    c    a       TRUE
    # 4    d    a    a    a    d    a       TRUE
    
    • 4
  5. lotus
    2024-01-08T22:25:57+08:002024-01-08T22:25:57+08:00

    Como foi observado, você deve evitar operações em linha. Aqui está uma alternativa que compara conjuntos usando uma função auxiliar que classifica com eficiência por linha para que as comparações sejam completamente vetorizadas.

    library(dplyr)
    
    f <- function(set1, set2) {
      s1 <- as.matrix(pick({{set1}}))
      s2 <- as.matrix(pick({{set2}}))
      row_sort <- function(m) matrix(m[order(row(m), m)], ncol = ncol(m), byrow = TRUE)
      !rowSums(row_sort(s1) != row_sort(s2)) > 0
    }
    
    dat %>%
      mutate(assessment = f(col1:col3, col4:col6))
    
      col1 col2 col3 col4 col5 col6 assessment
    1    a    a    a    a    a    b      FALSE
    2    b    b    a    b    a    b       TRUE
    3    c    d    a    d    c    a       TRUE
    4    d    a    a    a    d    a       TRUE
    
    • 4
  6. tmfmnk
    2024-01-08T21:48:15+08:002024-01-08T21:48:15+08:00

    Uma dplyropção vecsetspoderia ser:

    df %>%
     rowwise() %>%
     mutate(cond = vsetequal(c_across(col1:col3), c_across(col4:col6), multiple = TRUE))
    
      col1  col2  col3  col4  col5  col6  cond 
      <chr> <chr> <chr> <chr> <chr> <chr> <lgl>
    1 a     a     a     a     a     b     FALSE
    2 b     b     a     b     a     b     TRUE 
    3 c     d     a     d     c     a     TRUE 
    4 d     a     a     a     d     a     TRUE
    

    A mesma ideia com purrr::pmap():

    df %>%
     mutate(cond = pmap_lgl(across(col1:col6), 
                            ~ vsetequal(c(...)[1:3], c(...)[4:6], multiple = TRUE)))
    

    Esta é uma solução excessivamente ineficiente, mas por curiosidade:

    df %>%
     rowwise() %>%
     mutate(cond = toString(sort(c_across(col1:col3))) == toString(sort(c_across(col4:col6))))
    

    A mesma ideia com purrr:pmap():

    df %>%
     mutate(cond = pmap_lgl(across(col1:col6), 
                            ~ toString(sort(c(...)[1:3])) == toString(sort(c(...)[4:6]))))
    

    Usando a lógica de transposição do @SamR com vecsets:

    df %>%
     mutate(cond = map2_lgl(.x = across(col1:col3) %>% t() %>% data.frame(), 
                            .y = across(col4:col6) %>% t() %>% data.frame(), 
                            vsetequal))
    

    A mesma abordagem usando data.table::transpose():

    df %>%
     mutate(cond = map2_lgl(.x = data.table::transpose(across(col1:col3)), 
                            .y = data.table::transpose(across(col4:col6)), 
                            vsetequal))
    
    • 3
  7. ThomasIsCoding
    2024-01-08T22:33:51+08:002024-01-08T22:33:51+08:00

    Experimente o código abaixo com split.default+colMeans

    df$assessment <-
        colMeans(
            do.call(
                `==`,
                lapply(split.default(
                    df,
                    grepl("[1-3]$", names(df))
                ), \(d) apply(d, 1, sort))
            )
        ) == 1
    

    que deveria dar

    > df
      col1 col2 col3 col4 col5 col6 assessment
    1    a    a    a    a    a    b      FALSE
    2    b    b    a    b    a    b       TRUE
    3    c    d    a    d    c    a       TRUE
    4    d    a    a    a    d    a       TRUE
    
    • 3
  8. TarJae
    2024-01-08T23:41:33+08:002024-01-08T23:41:33+08:00

    Este é bastante detalhado, mas não consigo resistir. Aqui está um com pivotamento:

    library(dplyr)
    library(tidyr)
    
    df %>%
      pivot_longer(cols = starts_with("col"), names_to = "col_set") %>%
      group_by(group = (row_number() - 1) %/% ncol(df) + 1) %>% 
      mutate(x = lead(value, 3)) %>% 
      na.omit() %>% 
      mutate(across(c(value, x), ~sort(.))) %>% 
      summarize(check = all(value == x), .groups = "drop") %>% 
      bind_cols(df) %>% 
      select(-group)
    
    # A tibble: 4 × 7
      check col1  col2  col3  col4  col5  col6 
      <lgl> <chr> <chr> <chr> <chr> <chr> <chr>
    1 FALSE a     a     a     a     a     b    
    2 TRUE  b     b     a     b     a     b    
    3 TRUE  c     d     a     d     c     a    
    4 TRUE  d     a     a     a     d     a  
    
    • 3

relate perguntas

  • Adicionar número de série para atividade de cópia ao blob

  • A fonte dinâmica do empacotador duplica artefatos

  • Selecione linhas por grupo com 1s consecutivos

  • Lista de chamada de API de gráfico subscritoSkus estados Privilégios insuficientes enquanto os privilégios são concedidos

  • Função para criar DFs separados com base no valor da coluna

Sidebar

Stats

  • Perguntas 205573
  • respostas 270741
  • best respostas 135370
  • utilizador 68524
  • Highest score
  • respostas
  • Marko Smith

    Vue 3: Erro na criação "Identificador esperado, mas encontrado 'import'" [duplicado]

    • 1 respostas
  • Marko Smith

    Por que esse código Java simples e pequeno roda 30x mais rápido em todas as JVMs Graal, mas não em nenhuma JVM Oracle?

    • 1 respostas
  • Marko Smith

    Qual é o propósito de `enum class` com um tipo subjacente especificado, mas sem enumeradores?

    • 1 respostas
  • Marko Smith

    Como faço para corrigir um erro MODULE_NOT_FOUND para um módulo que não importei manualmente?

    • 6 respostas
  • Marko Smith

    `(expression, lvalue) = rvalue` é uma atribuição válida em C ou C++? Por que alguns compiladores aceitam/rejeitam isso?

    • 3 respostas
  • Marko Smith

    Quando devo usar um std::inplace_vector em vez de um std::vector?

    • 3 respostas
  • Marko Smith

    Um programa vazio que não faz nada em C++ precisa de um heap de 204 KB, mas não em C

    • 1 respostas
  • Marko Smith

    PowerBI atualmente quebrado com BigQuery: problema de driver Simba com atualização do Windows

    • 2 respostas
  • Marko Smith

    AdMob: MobileAds.initialize() - "java.lang.Integer não pode ser convertido em java.lang.String" para alguns dispositivos

    • 1 respostas
  • Marko Smith

    Estou tentando fazer o jogo pacman usando apenas o módulo Turtle Random e Math

    • 1 respostas
  • Martin Hope
    Aleksandr Dubinsky Por que a correspondência de padrões com o switch no InetAddress falha com 'não cobre todos os valores de entrada possíveis'? 2024-12-23 06:56:21 +0800 CST
  • Martin Hope
    Phillip Borge Por que esse código Java simples e pequeno roda 30x mais rápido em todas as JVMs Graal, mas não em nenhuma JVM Oracle? 2024-12-12 20:46:46 +0800 CST
  • Martin Hope
    Oodini Qual é o propósito de `enum class` com um tipo subjacente especificado, mas sem enumeradores? 2024-12-12 06:27:11 +0800 CST
  • Martin Hope
    sleeptightAnsiC `(expression, lvalue) = rvalue` é uma atribuição válida em C ou C++? Por que alguns compiladores aceitam/rejeitam isso? 2024-11-09 07:18:53 +0800 CST
  • Martin Hope
    The Mad Gamer Quando devo usar um std::inplace_vector em vez de um std::vector? 2024-10-29 23:01:00 +0800 CST
  • Martin Hope
    Chad Feller O ponto e vírgula agora é opcional em condicionais bash com [[ .. ]] na versão 5.2? 2024-10-21 05:50:33 +0800 CST
  • Martin Hope
    Wrench Por que um traço duplo (--) faz com que esta cláusula MariaDB seja avaliada como verdadeira? 2024-05-05 13:37:20 +0800 CST
  • Martin Hope
    Waket Zheng Por que `dict(id=1, **{'id': 2})` às vezes gera `KeyError: 'id'` em vez de um TypeError? 2024-05-04 14:19:19 +0800 CST
  • Martin Hope
    user924 AdMob: MobileAds.initialize() - "java.lang.Integer não pode ser convertido em java.lang.String" para alguns dispositivos 2024-03-20 03:12:31 +0800 CST
  • Martin Hope
    MarkB Por que o GCC gera código que executa condicionalmente uma implementação SIMD? 2024-02-17 06:17:14 +0800 CST

Hot tag

python javascript c++ c# java typescript sql reactjs html

Explore

  • Início
  • Perguntas
    • Recentes
    • Highest score
  • tag
  • help

Footer

AskOverflow.Dev

About Us

  • About Us
  • Contact Us

Legal Stuff

  • Privacy Policy

Language

  • Pt
  • Server
  • Unix

© 2023 AskOverflow.DEV All Rights Reserve