合并具有单元格不匹配的两个数据帧

问题描述 投票:0回答:1

我目前陷入困境:我有一个网站,为同一产品分配了不同的名称(即同一产品可能被列为“1”,“1号”,“3类 - 1号”)。对于这些值中的每一个,我想分配一个不同的结果(从具有更好名称的不同站点中删除)。

# Messy code from original website:
df1 <- data.frame(c(1,1,1,1,2,2,2,3,3,3,3,3,4,4,4), c("Number 1", "Number 2", 
                                                      "Category 1", "3", "8",
                                                      "Number 2 - Category 5","1", "Number 4", 
                                                      "Kat 1", "4", "Kat 2", 
                                                      "Number5", "Test", "4","3"))
colnames(df1) <- c("ID", "Category")

> df1
   ID              Category
1   1              Number 1
2   1              Number 2
3   1            Category 1
4   1                     3
5   2                     8
6   2 Number 2 - Category 5
7   2                     1
8   3              Number 4
9   3                 Kat 1
10  3                     4
11  3                 Kat 2
12  3               Number5
13  4                  Test
14  4                     4
15  4                     3

# Code from other site:
df2 <- data.frame(c(1,1,1,1,2,2,2,2,3,3,3,3,3,4,4,4,4,4), c("1", "2", "3", "Category 1",
                                                      "Category 5", "1", "2", "3", "4",
                                                      "Kat 1", "Kat 2", "Kat 3","5",
                                                      "1","2","3","4","Test"), 
                  c(1,2,3,4,5,6,5,8,9,10,11,12,13,14,15,16,17,18))
colnames(df2) <- c("ID", "Category", "Outcome")

> df2
   ID   Category Outcome
1   1          1       1
2   1          2       2
3   1          3       3
4   1 Category 1       4
5   2 Category 5       5
6   2          1       6
7   2          2       5
8   2          3       8
9   3          4       9
10  3      Kat 1      10
11  3      Kat 2      11
12  3      Kat 3      12
13  3          5      13
14  4          1      14
15  4          2      15
16  4          3      16
17  4          4      17
18  4       Test      18

我想保留数据帧1,但是指定从df2获取的“输出”,由ID和类别匹配。这可以通过左连接轻松完成,但正如您所看到的,类别级别不同。

我一直在考虑重命名级别,但考虑到我的数据集的大小(~500,000行),这似乎不可行,我希望有一个更简单的(grepl,gsub?)解决方案。

我的最终结果应如下所示:

> df1
   ID              Category Outcome
1   1              Number 1       1
2   1              Number 2       2
3   1            Category 1       4
4   1                     3       3
5   2                     8      NA
6   2 Number 2 - Category 5       5
7   2                     1       6
8   3              Number 4       9
9   3                 Kat 1      10
10  3                     4       9
11  3                 Kat 2      11
12  3               Number5      13
13  4                  Test      18
14  4                     4      17
15  4                     3      16

很抱歉这个复杂的例子,我想确保涵盖所有意外情况。这里有一点解释:

每当它有一个类别时,通常会以完整的方式列出正确的拼写(即“Kat”将始终以“Kat”出现,“Category”将始终以“Category”出现)。数字可能出现在前面的“数字”一词或简单地作为数字本身。如果没有出现数字,我希望它显示NA - 我将不得不手动添加缺失的数字。

非常感谢您的阅读 - 非常感谢您的帮助!

r dataframe merge gsub grepl
1个回答
0
投票

这可能会大大简化,但您可以尝试:

library(dplyr)

extr_last_digit <- function(x) substring(x, first = regexpr("(\\d+)(?!.*\\d)", x, perl = T))
extr_cat_digit <- function(x) gsub(".*((?<=Category\\s)\\d+).*|.*((?<=Kat\\s)\\d+).*", "\\1\\2", x, perl = T)

df1 %>%
  mutate(
    join = case_when(
      !grepl("Category|Kat", Category) ~ extr_last_digit(Category),
      TRUE ~ paste0("Category ", extr_cat_digit(Category)))
  ) %>%
  left_join(df2 %>%
              mutate(
                join = case_when(
                  !grepl("Category|Kat", Category) ~ extr_last_digit(Category),
                  TRUE ~ paste0("Category ", extr_cat_digit(Category))
                ),
                Category = NULL), 
            by = c("ID", "join")
  ) %>%
  select(-join)

输出:

   ID              Category Outcome
1   1              Number 1       1
2   1              Number 2       2
3   1            Category 1       4
4   1                     3       3
5   2                     8      NA
6   2 Number 2 - Category 5       5
7   2                     1       6
8   3              Number 4       9
9   3                 Kat 1      10
10  3                     4       9
11  3                 Kat 2      11
12  3               Number5      13
13  4                  Test      18
14  4                     4      17
15  4                     3      16

基本上它的作用是它在两个具有相同结构的join中创建一个df列 - 如果它没有找到任何提及CategoryKat,它使用最后提到的数字进行合并,否则它粘贴单词Category和数字在CategoryKat之后提到。

这也适用于Test这个词,因为如果它找不到任何数字,它只使用整个字符串。

© www.soinside.com 2019 - 2024. All rights reserved.