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 are not supported and produce an error.

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(1103)
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] (402,1.13e+03]  (1.13e+03, Inf] (149,402]       (402,1.13e+03] 
#> [5] (149,402]       [-Inf,149]     
#> Levels: [-Inf,149] (149,402] (402,1.13e+03] (1.13e+03, Inf]

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