Subset data frame based on number of rows per group

by

First, two base alternatives. One relies on table, and the other on ave and length. Then, two data.table ways.

1. table

tt <- table(df$name)

df2 <- subset(df, name %in% names(tt[tt < 3]))
# or
df2 <- df[df$name %in% names(tt[tt < 3]), ]

If you want to walk it through step by step:

# count each 'name', assign result to an object 'tt'
tt <- table(df$name)

# which 'name' in 'tt' occur more than three times?
# Result is a logical vector that can be used to subset the table 'tt'
tt < 3

# from the table, select 'name' that occur < 3 times
tt[tt < 3]

# ...their names
names(tt[tt < 3])

# rows of 'name' in the data frame that matches "the < 3 names"
# the result is a logical vector that can be used to subset the data frame 'df'
df$name %in% names(tt[tt < 3])

# subset data frame by a logical vector
# 'TRUE' rows are kept, 'FALSE' rows are removed.
# assign the result to a data frame with a new name
df2 <- subset(df, name %in% names(tt[tt < 3]))
# or
df2 <- df[df$name %in% names(tt[tt < 3]), ]

2. ave and length

As suggested by @flodel:

df[ave(df$x, df$name, FUN = length) < 3, ]

3. data.table: .N and .SD:

library(data.table)
setDT(df)[, if (.N < 3) .SD, by = name]

4. data.table: .N and .I:

setDT(df)
df[df[, .I[.N < 3], name]$V1]

See also the related Q&A Count number of observations/rows per group and add result to data frame.

More Related Contents:

  1. Filter data.frame rows by a logical condition
  2. Subset / filter rows in a data frame based on a condition in a column
  3. Reshaping data.frame from wide to long format
  4. How to sum a variable by group
  5. How do I make a list of data frames?
  6. Aggregate / summarize multiple variables per group (e.g. sum, mean)
  7. Sort (order) data frame rows by multiple columns
  8. Split data frame string column into multiple columns
  9. Quickly reading very large tables as dataframes
  10. Why is `[` better than `subset`?
  11. Pass a data.frame column name to a function
  12. Drop data frame columns by name
  13. How to subset matrix to one column, maintain matrix data type, maintain row/column names?
  14. Sample random rows in dataframe
  15. Extracting specific columns from a data frame
  16. How to drop columns by name in a data frame
  17. How does one reorder columns in a data frame?
  18. How do I extract a single column from a data.frame as a data.frame?
  19. Subset dataframe by multiple logical conditions of rows to remove
  20. Filtering a data frame on a vector [duplicate]
  21. Combine two or more columns in a dataframe into a new column with a new name
  22. Subset rows in a data frame based on a vector of values
  23. Subset of rows containing NA (missing) values in a chosen column of a data frame
  24. How to subset data in R without losing NA rows?
  25. Using grep to help subset a data frame
  26. Convert a dataframe to a vector (by rows)
  27. Filter data frame rows based on values in vector
  28. Should I use a data.frame or a matrix?
  29. Selecting columns in R data frame based on those *not* in a vector
  30. Return df with a columns values that occur more than once [duplicate]

Leave a Comment