首页 > 解决方案 > 在不使用循环的情况下创建所需的小标题

问题描述

我在数据中有 n 个分段(0 到 n-1)。我知道每个细分市场中男性的百分比。

如何在不使用循环的情况下编写动态代码来创建分段和男性数组?

例如 -

# datasize = 20, # segmentations = 2
N = 20
percent_each_segmentation = c(0.4, 0.6)
percent_male_per_segmentation = c(0.5, 0.25)

所需的输出看起来像

segmentation <- c(rep(0, 8), rep(1, 12))
# segmentation 0 - 0.4*20, segmentation 1 - 0.6*20

male <- c(rep(1, 4), rep(0, 4), rep(1, 3), rep(0, 9))
# male in segmentation 0 : 0.5*0.4*20, segmentation 1 : 0.25*0.6*20

tbl <- tibble(segmentation, male)

标签: rdplyrtidyversetibble

解决方案


我们可以使用repas

N <- 20
r1 <- percent_each_segmentation * 
      percent_male_per_segmentation * N
bnry <- c(1, 0)
r2 <- rep(r1, each = length(bnry))

r2[length(r2)] <- N - sum(head(r2, -1))


rep(rep(bnry, length.out = length(r2)), r2)

对于分割

rep(c(0, 1), N * percent_each_segmentation)

推荐阅读