Stratified random sampling from data frame

by

I would suggest using either stratified from my “splitstackshape” package, or sample_n from the “dplyr” package:

## Sample data
set.seed(1)
n <- 1e4
d <- data.table(age = sample(1:5, n, T), 
                lc = rbinom(n, 1 , .5),
                ants = rbinom(n, 1, .7))
# table(d$age, d$lc)

For stratified, you basically specify the dataset, the stratifying columns, and an integer representing the size you want from each group OR a decimal representing the fraction you want returned (for example, .1 represents 10% from each group).

library(splitstackshape)
set.seed(1)
out <- stratified(d, c("age", "lc"), 30)
head(out)
#    age lc ants
# 1:   1  0    1
# 2:   1  0    0
# 3:   1  0    1
# 4:   1  0    1
# 5:   1  0    0
# 6:   1  0    1

table(out$age, out$lc)
#    
#      0  1
#   1 30 30
#   2 30 30
#   3 30 30
#   4 30 30
#   5 30 30

For sample_n you first create a grouped table (using group_by) and then specify the number of observations you want. If you wanted proportional sampling instead, you should use sample_frac.

library(dplyr)
set.seed(1)
out2 <- d %>%
  group_by(age, lc) %>%
  sample_n(30)

# table(out2$age, out2$lc)

More Related Contents:

  1. Sample random rows in dataframe
  2. Reasons for using the set.seed function
  3. Generate N random integers that sum to M in R
  4. efficiently generate a random sample of times and dates between two dates
  5. Sample n random rows per group in a dataframe
  6. Generate random numbers with fixed mean and sd
  7. How to randomize (or permute) a dataframe rowwise and columnwise?
  8. from data table, randomly select one row per group
  9. Generating Random Strings
  10. Generate correlated random numbers from binomial distributions
  11. How do I best simulate an arbitrary univariate random variate using its probability function?
  12. Local Variables Within aes
  13. dplyr – using column names as function arguments
  14. Making a stacked bar plot for multiple variables – ggplot2 in R
  15. read.csv warning ‘EOF within quoted string’ prevents complete reading of file
  16. Merge dataframes of different sizes
  17. Extract last word in string in R
  18. Standard Deviation in R Seems to be Returning the Wrong Answer – Am I Doing Something Wrong?
  19. Split Data Frame into Rows of Fixed Size
  20. How to tell lapply to ignore an error and process the next thing in the list?
  21. Can ggplot2 control point size and line size (lineweight) separately in one legend?
  22. How does one stop using rowwise in dplyr?
  23. What is the benefit of import in a namespace in R?
  24. R DBI ODBC error: nanodbc/nanodbc.cpp:3110: 07009: [Microsoft][ODBC Driver 13 for SQL Server]Invalid Descriptor Index
  25. Subsetting a matrix by row.names
  26. R divide data into groups
  27. Configuring compilers on Apple silicon (Big Sur, Monterey, Ventura) for Rcpp and other tools
  28. Creating a vertical color gradient for a geom_bar plot
  29. Divide each value by the sum of values by group
  30. Dynamic linking with rpath not working under Ubuntu 17.10

Leave a Comment