使用data.table删除R中的行

时间:2016-08-22 17:30:02

标签: r data.table

DF: -

Date    Name  Salary 
Q1 2015 ABC   $10
Q2 2015 ABC   $11
Q3 2015 ABC   $15
Q1 2015 XYZ   $25
Q2 2015 XYZ   $20

我想从总频率小于3的数据中删除行。例如XYZ的频率为2,因此我想删除第4行和第5行。

test <- setDT(df)[,.I[.N>2],by=Name]

输出: -

> test
   Name V1
1:  ABC  1
2:  ABC  2
3:  ABC  3

过滤正确完成但我没有得到整个数据集,我只在输出中获得了Name列。

1 个答案:

答案 0 :(得分:4)

我们需要提取'V1'列并将其用作'i'中的行索引来对行进行子集化。

setDT(df)[df[,.I[.N>2],by=Name]$V1]
#       Date Name Salary
#1: Q1 2015  ABC    $10
#2: Q2 2015  ABC    $11
#3: Q3 2015  ABC    $15

if.SD

的简明选项
setDT(df)[, if(.N >2) .SD, by = Name]
#    Name    Date Salary
#1:  ABC Q1 2015    $10
#2:  ABC Q2 2015    $11
#3:  ABC Q3 2015    $15

以防万一,如果我们需要dplyr方法

library(dplyr)
df %>%
   group_by(Name) %>%
   filter(n() >2 )
#      Date  Name Salary
#     <chr> <chr>  <chr>
#1 Q1 2015   ABC    $10
#2 Q2 2015   ABC    $11
#3 Q3 2015   ABC    $15

或者使用base R,我们可以选择多个选项,其中一个选项ave

df[with(df, ave(seq_along(Name), Name, FUN = length)>2),]

或使用table

tbl <- table(df$Name)> 2
subset(df, Name %in% names(tbl)[tbl])