如何将 combn() 函数应用于列表类型的列？答案

【问题标题】：How to apply the combn() function to a column of list type?如何将 combn() 函数应用于列表类型的列？
【发布时间】：2021-05-18 15:41:23
【问题描述】：

我有一个包含 3 列的数据框，其中一列是列表类型

>head(basket_data)
# A tibble: 8 x 3
  order_id items      count
     <int> <list>     <int>
1        2 <chr [9]>      9
2        3 <chr [8]>      8
3        4 <chr [13]>    13
4        5 <chr [26]>    26
5        6 <chr [3]>      3

说第 1 行（order_id 2）的列表是 [a,b,c]

我想使用 combn 函数（也许？）创建一个新列，它将仅包含该行列表的所有对组合，所以 [[a,b],[b,c],[a,c ]]

我在 SO 上遇到的所有使用 combn 的示例都将数据框中的每个列表组合在一起，并将整个列表配对。任何帮助，将不胜感激。谢谢！

【问题讨论】：

你能提供dput(head(basket_data))吗？

标签： r combn

【解决方案1】：

假设您的数据是这样的

test <- structure(list(items = list(c('a', 'b'), c('b', 'c', 'd'), c('d', 'e'), c('f', 'g', 'i'), c('g', 'h')), 
               ID = 1:5), row.names = c(NA, 5L), class = "data.frame")

test
    items ID
1    a, b  1
2 b, c, d  2
3    d, e  3
4 f, g, i  4
5    g, h  5

as_tibble(test)
# A tibble: 5 x 2
  items        ID
  <list>    <int>
1 <chr [2]>     1
2 <chr [3]>     2
3 <chr [2]>     3
4 <chr [3]>     4
5 <chr [2]>     5

那么你可以这样做

as_tibble(test) %>% mutate(combs = map(items, ~combn(.x, 2)))

# A tibble: 5 x 3
  items        ID combs            
  <list>    <int> <list>           
1 <chr [2]>     1 <chr[,1] [2 x 1]>
2 <chr [3]>     2 <chr[,3] [2 x 3]>
3 <chr [2]>     3 <chr[,1] [2 x 1]>
4 <chr [3]>     4 <chr[,3] [2 x 3]>
5 <chr [2]>     5 <chr[,1] [2 x 1]>

检查

as_tibble(test) %>% mutate(combs = map(items, ~combn(.x, 2))) %>%
  data.frame()
    items ID            combs
1    a, b  1             a, b
2 b, c, d  2 b, c, b, d, c, d
3    d, e  3             d, e
4 f, g, i  4 f, g, f, i, g, i
5    g, h  5             g, h

或

as_tibble(test) %>% mutate(combs = map(items, ~combn(.x, 2, list)))

# A tibble: 5 x 3
  items        ID combs     
  <list>    <int> <list>    
1 <chr [2]>     1 <list [1]>
2 <chr [3]>     2 <list [3]>
3 <chr [2]>     3 <list [1]>
4 <chr [3]>     4 <list [3]>
5 <chr [2]>     5 <list [1]>

取决于你想要的输入和输出格式

【讨论】：

【解决方案2】：

这里尝试使用dplyr 和purrr 以及使用您的共享示例随机生成的一些示例数据。

library(dplyr)
library(purrr)

set.seed(10)
basket_data <- tibble(
  order_id = seq(2, 6, by = 1),
  items = lapply(floor(runif(5, 5, 20)),
    FUN = function(x) { sample(letters, size = x) }),
  count = floor(runif(5, 1, 30))
)

basket_data$new_col <- map(basket_data$items,
  .f = function(x) combn(x, 2, FUN = function(x) list(x)))

basket_data
#> # A tibble: 5 x 4
#>   order_id items      count new_col     
#>      <dbl> <list>     <dbl> <list>      
#> 1        2 <chr [12]>    24 <list [66]> 
#> 2        3 <chr [9]>      8 <list [36]> 
#> 3        4 <chr [11]>     5 <list [55]> 
#> 4        5 <chr [15]>     5 <list [105]>
#> 5        6 <chr [6]>     15 <list [15]>

一些结果样本

# Here is first items list
basket_data$items[[1]]
#>  [1] "w" "h" "v" "g" "s" "o" "u" "j" "z" "x" "b" "y"

# Here is some example of new_col for first items
basket_data$new_col[[1]][1:10]
#> [[1]]
#> [1] "w" "h"
#> 
#> [[2]]
#> [1] "w" "v"
#> 
#> [[3]]
#> [1] "w" "g"
#> 
#> [[4]]
#> [1] "w" "s"
#> 
#> [[5]]
#> [1] "w" "o"
#> 
#> [[6]]
#> [1] "w" "u"
#> 
#> [[7]]
#> [1] "w" "j"
#> 
#> [[8]]
#> [1] "w" "z"
#> 
#> [[9]]
#> [1] "w" "x"
#> 
#> [[10]]
#> [1] "w" "b"

^{由reprex package (v2.0.0) 于 2021 年 5 月 18 日创建}

【讨论】：