Início / coding / Perguntas / 79437665
Accepted
user27247029
Asked: 2025-02-14 04:39:49 +0800 CST

Como obter todas as combinações de linhas em um data.frame mantendo os valores na segunda coluna para qualquer número de combinações?

  • 772

Estou tentando obter todas as combinações de linhas de uma coluna para ela mesma, mantendo os valores de uma segunda coluna.

library(dplyr)
library(tidyr)

dt0 <-
  data.frame(
    row = letters[1:10],
    n1 = c(2, 2, 1, 3, 1, 5, 1, 3, 2, 2)
  )

dt0 |>
  expand(
    row1 = row,
    row2 = row
  ) |>
  filter(row1 < row2) |>
  left_join(
    dt0 |>
      rename(n1.x = n1),
    by = join_by(row1 == row)
  ) |>
  left_join(
    dt0 |>
      rename(n1.y = n1),
    by = join_by(row2 == row)
  )

o resultado esperado é:

# A tibble: 45 × 4
   row1  row2   n1.x  n1.y
   <chr> <chr> <dbl> <dbl>
 1 a     b         2     2
 2 a     c         2     1
 3 a     d         2     3
 4 a     e         2     1
 5 a     f         2     5
 6 a     g         2     1
 7 a     h         2     3
 8 a     i         2     2
 9 a     j         2     2
10 b     c         2     1
# ℹ 35 more rows
# ℹ Use `print(n = ...)` to see more rows

Mas não sei como generalizar isso para gerar todas as combinações dos elementos das linhas do data.frame tomadas mde uma vez, então minha pergunta é: Como posso generalizar esse padrão para qualquer número de linhas em expand(...)? Por exemplo, com três

dt0 |>
  expand(
    row1 = row,
    row2 = row,
    row3 = row
  ) |>
  filter(row1 < row2) |>
  filter(row2 < row3) |>
  left_join(
    dt0 |>
      rename(n1.x = n1),
    by = join_by(row1 == row)
  ) |>
  left_join(
    dt0 |>
      rename(n1.y = n1),
    by = join_by(row2 == row)
  ) |>
  left_join(
    dt0 |>
      rename(n1.z = n1),
    by = join_by(row3 == row)
  )

# A tibble: 120 × 6
   row1  row2  row3   n1.x  n1.y  n1.z
   <chr> <chr> <chr> <dbl> <dbl> <dbl>
 1 a     b     c         2     2     1
 2 a     b     d         2     2     3
 3 a     b     e         2     2     1
 4 a     b     f         2     2     5
 5 a     b     g         2     2     1
 6 a     b     h         2     2     3
 7 a     b     i         2     2     2
 8 a     b     j         2     2     2
 9 a     c     d         2     1     3
10 a     c     e         2     1     1
# ℹ 110 more rows
# ℹ Use `print(n = ...)` to see more rows

2 respostas

  1. Best Answer

    Acho combnque expandse encaixa melhor no seu propósito

    f <- function(dt0, k) {
    with(
        dt0,
        cbind(
            setNames(data.frame(t(combn(row, k))), paste0("row", seq(k))),
            setNames(data.frame(t(combn(n1, k))), paste0("n1.", seq(k)))
        )
    )
}

    ou

    f <- function(dt0, k) {
    do.call(
        cbind,
        lapply(
            seq_along(dt0),
            \(i) setNames(
                data.frame(t(combn(dt0[[i]], k))),
                paste0(names(dt0[i]), ".", seq(k))
            )
        )
    )
}

    tal que

    > head(f(dt0, 2), 10)
   row.1 row.2 n1.1 n1.2
1      a     b    2    2
2      a     c    2    1
3      a     d    2    3
4      a     e    2    1
5      a     f    2    5
6      a     g    2    1
7      a     h    2    3
8      a     i    2    2
9      a     j    2    2
10     b     c    2    1

> head(f(dt0, 3), 10)
   row.1 row.2 row.3 n1.1 n1.2 n1.3
1      a     b     c    2    2    1
2      a     b     d    2    2    3
3      a     b     e    2    2    1
4      a     b     f    2    2    5
5      a     b     g    2    2    1
6      a     b     h    2    2    3
7      a     b     i    2    2    2
8      a     b     j    2    2    2
9      a     c     d    2    1    3
10     a     c     e    2    1    1
    • 5

  2. Você pode usar tidyverse:

    library(tidyverse)
n <- 2
l <- rep(list(as_tibble(dt0)), n)
reduce(l, ~left_join(.x, .y, join_by(row < row), suffix = c("x", "")))

A tibble: 46 × 4
   rowx    n1x row      n1
   <chr> <dbl> <chr> <dbl>
 1 a         2 b         2
 2 a         2 c         1
 3 a         2 d         3
 4 a         2 e         1
 5 a         2 f         5
 6 a         2 g         1
 7 a         2 h         3
 8 a         2 i         2
 9 a         2 j         2
10 b         2 c         1

    Para generalizar:

    fn <- function(df, n){
  s <- str_c("_", seq_len(n-1))
  l <- rep(list(as_tibble(df)), n)
  reduce2(l, s, ~left_join(.x, .y, join_by(row < row), suffix = c(..3, "")))
}

fn(dt0, 4)
# A tibble: 256 × 8
   row_1  n1_1 row_2  n1_2 row_3  n1_3 row      n1
   <chr> <dbl> <chr> <dbl> <chr> <dbl> <chr> <dbl>
 1 a         2 b         2 c         1 d         3
 2 a         2 b         2 c         1 e         1
 3 a         2 b         2 c         1 f         5
 4 a         2 b         2 c         1 g         1
 5 a         2 b         2 c         1 h         3
 6 a         2 b         2 c         1 i         2
 7 a         2 b         2 c         1 j         2
 8 a         2 b         2 d         3 e         1
 9 a         2 b         2 d         3 f         5
10 a         2 b         2 d         3 g         1
# ℹ 246 more rows
# ℹ Use `print(n = ...)` to see more rows
    • 2

relate perguntas