填充 matrix/data 框架的 tidyverse 解决方案

Question

我有一个包含单列的数据框（命中）。这里填充了独特的搜索结果。

第二个数据框（数据）包含各种搜索查询的结果。列名标识使用的搜索词，行填充搜索结果。

我想构建一个矩阵或另一个根据搜索结果是否按列填充的数据框。

我可以使用 base R 和以下代码来做到这一点：

    library(tidyverse)
    hit <- read_csv("hit
                     A1 
                     A3
                     B2
                     B4
                     D3")

    data <- read_csv("Search1, Search2, Search3, Search4
             A1, B4, A3, A1
             B4, D3, NA, B2
             D3, NA, NA, B4")



    search <- c("Search1", "Search2", "Search3", "Search4")

    the_matrix <- matrix(data = NA, nrow = 5, ncol = 4)
    rownames(the_matrix) <- hit$hit 
    colnames(the_matrix) <- search

    for (i in search)
        for (j in 1:3){
            result <- data[[i]][[j]]
            row_index <- which(rownames(the_matrix) == result)
            the_matrix[row_index, i] <- 1
        }

    the_matrix[is.na(the_matrix)] <- 0

在我看来，应该有一种方法可以使用 tidyverse 实现相同的结果，使用第一个数据帧作为起点。从那里开始，使用搜索结果作为要填充的键逐列引入第二个数据框。

有人可以帮忙吗？

Answer 1

data %>% gather(na.rm=T) %>% mutate(p=1L) %>% spread("key", "p", fill=0L)

# A tibble: 5 x 5
  value Search1 Search2 Search3 Search4
  <chr>   <int>   <int>   <int>   <int>
1 A1          1       0       0       1
2 A3          0       0       1       0
3 B2          0       0       0       1
4 B4          1       1       0       1
5 D3          1       1       0       0

Answer 2

您可以将 map_df 与 match 结合使用，然后将 a_tibble 中的所有非 0 替换为 1L。

library(purrr)
library(tidyr)
a_tibble <- map_df(data, ~match(hit[["hit"]], ., nomatch = 0L))
a_tibble[a_tibble != 0] <- 1L
a_tibble %>%
  add_column(., hit = hit$hit, .before = 1)
#  A tibble: 5 x 4
#  hit   Search1 Search2 Search3 Search4
#  <chr>   <int>   <int>   <int>   <int>
#1 A1          1       0       0       1
#2 A3          0       0       1       0
#3 B2          0       0       0       1
#4 B4          1       1       0       1
#5 D3          1       1       0       0

Answer 3

供参考，您还可以有一个相当优雅的 base r 解决方案

the_matrix=sapply(data,function(x) as.numeric(hit$hit%in%x))
rownames(the_matrix)<-hit$hit

填充 matrix/data 框架的 tidyverse 解决方案

A tidyverse solution to populating a matrix/data frame

join

r

dplyr

tidyverse