Skip to contents

Apply strata boundaries from a strata_bound() result to a numeric vector, returning a factor of stratum assignments.

Usage

# S3 method for class 'svyplan_strata'
predict(object, newdata, labels = NULL, ...)

Arguments

object

A svyplan_strata object.

newdata

Numeric vector to stratify.

labels

Labels for the resulting factor levels. Default NULL generates labels from the boundary intervals.

...

Additional arguments (currently unused).

Value

A factor of stratum assignments with length equal to length(newdata). Values beyond the original training range are assigned to the lowest or highest stratum.

See also

strata_bound() to compute the boundaries.

Examples

set.seed(42)
x <- rlnorm(500, meanlog = 6, sdlog = 1.5)
sb <- strata_bound(x, n_strata = 4, n = 100)

# Default interval labels
head(predict(sb, x))
#> [1] (2.09e+03, Inf] [-Inf,295]      (295,828]       (828,2.09e+03] 
#> [5] (295,828]       (295,828]      
#> Levels: [-Inf,295] (295,828] (828,2.09e+03] (2.09e+03, Inf]

# Custom labels
predict(sb, x, labels = c("Low", "Mid-Low", "Mid-High", "High"))
#>   [1] High     Low      Mid-Low  Mid-High Mid-Low  Mid-Low  High     Mid-Low 
#>   [9] High     Mid-Low  High     High     Low      Low      Mid-Low  Mid-High
#>  [17] Low      Low      Low      High     Low      Low      Mid-Low  High    
#>  [25] High     Low      Low      Low      Mid-Low  Low      Mid-Low  Mid-High
#>  [33] Mid-High Low      Mid-High Low      Low      Low      Low      Mid-Low 
#>  [41] Mid-Low  Low      Mid-High Low      Low      Mid-Low  Low      High    
#>  [49] Low      Mid-High Mid-Low  Low      High     Mid-High Mid-Low  Mid-Low 
#>  [57] Mid-High Mid-Low  Low      Mid-Low  Low      Mid-Low  Mid-High High    
#>  [65] Low      High     Mid-Low  Mid-High Mid-High Mid-High Low      Mid-Low 
#>  [73] Mid-High Low      Low      Mid-High Mid-High Mid-Low  Low      Low     
#>  [81] High     Mid-Low  Mid-Low  Mid-Low  Low      Mid-High Low      Mid-Low 
#>  [89] Mid-High Mid-High High     Low      Mid-High High     Low      Low     
#>  [97] Low      Low      Mid-Low  Mid-High High     Mid-High Low      High    
#> [105] Low      Mid-Low  Low      Mid-Low  Mid-Low  Mid-Low  Mid-Low  Mid-Low 
#> [113] Low      Low      Low      Low      Low      High     Low      Mid-Low 
#> [121] Low      Low      Mid-Low  Low      Mid-Low  Low      Low      Low     
#> [129] Low      Mid-Low  Mid-High Low      Mid-Low  High     High     Low     
#> [137] Mid-Low  High     Low      Mid-Low  Mid-Low  Low      Low      Mid-Low 
#> [145] Low      High     Low      Low      Mid-High Low      Mid-Low  Low     
#> [153] High     Low      Low      Low      Mid-Low  Low      Low      High    
#> [161] Mid-Low  Low      Mid-Low  Low      Mid-High High     Low      Mid-Low 
#> [169] Mid-Low  Mid-High Low      Mid-High Low      High     Mid-High Mid-Low 
#> [177] Low      Mid-High Mid-High Mid-Low  Mid-Low  Low      Mid-Low  Mid-Low 
#> [185] Low      Low      Mid-High Mid-High Low      Low      Mid-Low  Low     
#> [193] Mid-Low  Low      Low      Mid-High Mid-Low  Mid-High High     Mid-Low 
#> [201] Low      Mid-Low  High     High     Low      Low      Low      Low     
#> [209] Low      Mid-Low  Low      High     Mid-Low  Mid-Low  Mid-High Mid-Low 
#> [217] Mid-Low  High     Low      Low      Mid-Low  Low      Low      Low     
#> [225] Mid-Low  Mid-Low  Mid-High Low      Low      High     Low      Mid-High
#> [233] Low      Low      Low      Low      High     Mid-Low  Mid-Low  Low     
#> [241] Low      Mid-High High     High     Low      High     Mid-High Mid-Low 
#> [249] Mid-High Low      Low      Mid-Low  Low      Mid-Low  High     Low     
#> [257] Low      Mid-Low  Low      Low      Mid-High Mid-High Mid-Low  Low     
#> [265] Mid-Low  Mid-High Mid-High Mid-Low  Low      Mid-Low  Mid-High Mid-Low 
#> [273] Mid-Low  Mid-Low  Low      High     Mid-Low  Low      High     High    
#> [281] Mid-High Low      Low      Mid-High Mid-Low  Mid-Low  High     Low     
#> [289] Low      Mid-Low  Low      Mid-Low  Low      High     Mid-Low  Low     
#> [297] Mid-Low  Mid-High Mid-Low  Low      Mid-Low  Mid-High Mid-Low  Mid-High
#> [305] Mid-Low  Mid-Low  Mid-High Mid-High Low      Mid-Low  Mid-Low  Low     
#> [313] Low      Low      Mid-High Low      Low      Mid-Low  Low      Mid-High
#> [321] Low      Mid-Low  Mid-Low  Mid-Low  High     Low      High     Mid-Low 
#> [329] Low      Mid-Low  Mid-High Low      Low      Mid-Low  Mid-High Mid-Low 
#> [337] Low      Low      Low      Low      Mid-Low  Mid-High Low      Low     
#> [345] Mid-High Mid-Low  High     Low      Mid-Low  High     Mid-High Low     
#> [353] Mid-High Low      Low      High     Low      Mid-High Low      Low     
#> [361] Low      Low      Low      Low      Mid-Low  Low      High     Low     
#> [369] Low      Mid-Low  High     Mid-Low  High     Low      High     Low     
#> [377] Low      Low      Mid-Low  High     High     Low      Mid-High High    
#> [385] Mid-Low  Low      Low      Mid-High High     Mid-High Mid-Low  Low     
#> [393] Mid-High Low      Low      Mid-High Mid-Low  Mid-Low  Low      Mid-High
#> [401] High     Low      Mid-Low  Mid-Low  Low      Low      Mid-Low  Mid-High
#> [409] High     Low      Low      Low      Low      Low      Low      Low     
#> [417] Low      Mid-Low  Low      High     Mid-High Low      Mid-Low  Low     
#> [425] Mid-High Mid-Low  Low      Mid-Low  Mid-Low  Mid-Low  High     Low     
#> [433] Low      Low      Mid-Low  Mid-High Low      Low      Low      High    
#> [441] High     Mid-Low  High     Low      Mid-High Low      Mid-Low  Low     
#> [449] Low      Mid-High Low      Low      Mid-Low  Low      Mid-Low  Low     
#> [457] High     Mid-High High     Low      Mid-High High     Mid-Low  Low     
#> [465] Low      Mid-Low  High     Mid-High Low      Low      Mid-Low  Low     
#> [473] Mid-High Low      Mid-Low  Low      Low      Low      Mid-Low  Mid-Low 
#> [481] Low      Low      Low      Low      Mid-High Low      Mid-Low  Low     
#> [489] Low      Mid-High Mid-Low  Mid-Low  Mid-Low  High     Low      Mid-High
#> [497] Mid-High Low      Mid-Low  Mid-Low 
#> Levels: Low Mid-Low Mid-High High