在数据框中提取具有重复对的不同行的列位置

时间:2019-01-04 13:09:24

标签: r

我的数据框如下:

master <- data.frame(A=c(1,1,2,2,3,3,4,4,5,5), B=c(1,2,3,3,4,5,6,6,7,8),C=c(5,2,5,7,7,5,7,9,7,8),D=c(1,2,5,3,7,5,9,6,7,0))

如您所见,我有4列A,B,C,D。我要实现的是创建一个新的数据框,其中将包含AB之间的重复对行,发生重复的行的索引,使这些行不同的列名( C,D,或CD)在第三列中,以及使这些行不同的那些列名的列位置。

所需的输出应类似于:

# A tibble: 2 x 4
# Groups:   A [?]
#     A     B     n locs  different position
#   <dbl> <dbl> <int> <chr> <chr>    <int>
#1     2     3     2 3, 4  C, D     3,4
#2     4     6     2 7, 8  C, D     3,4

如您所见,重复的成对行的区别在于C列和D列分别位于主数据集的第3和第4r位置。正如您在下面看到的,唯一需要添加的是基于初始master数据集的列位置。

#code
library(tidyverse)
master %>%
   mutate(rn = row_number()) %>% 
   group_by(A, B) %>% 
   filter(n() > 1) %>% 
   summarise(n = n(), 
            locs =  toString(rn),
            Different = case_when(n_distinct(C) > 1 & n_distinct(D) > 1 ~ 'C, D', 
                                  n_distinct(C) > 1 ~ 'C', 
                                  n_distinct(D) > 1 ~ 'D', 
                                  TRUE ~ 'Same')) 

# A tibble: 2 x 4
# Groups:   A [?]
#     A     B     n locs  different
#   <dbl> <dbl> <int> <chr> <chr>    
#1     2     3     2 3, 4  C, D     
#2     4     6     2 7, 8  C, D 

1 个答案:

答案 0 :(得分:2)

您可以使用data.table,它具有方便的符号.N用于组大小,.I用于行号。剩下的就是检查哪些列不同,您可以通过sapply将它们覆盖并用uniqueN检查唯一元素的数量来完成。

library(data.table)
library(magrittr)
setDT(master)

out <- 
  master[, .(n = .N, 
             locs = .(.I), 
             different = .(names(.SD)[sapply(.SD, uniqueN) != 1]))
         , by = .(A, B)] %>% 
    .[n > 1]

out
#    A B n locs different
# 1: 2 3 2  3,4       C,D
# 2: 4 6 2  7,8       C,D

然后,您可以使用position来获得match(different, names(master)),并使用lapply一次一行。

out[, position := lapply(different, match, names(master))]

out
#    A B n locs different position
# 1: 2 3 2  3,4       C,D      3,4
# 2: 4 6 2  7,8       C,D      3,4