在R中,我如何在局部混洗矢量元素

时间:2013-07-14 15:09:04

标签: r sample

我在R中有以下向量。将它们视为数字向量。

x = c(1,2,3,4,...100)

我想根据一些输入数字“局部因子”将这个矢量“本地”随机化。例如,如果位置因子是3,那么前3个元素被采用并随机化,接着是接下来的3个元素,依此类推。有没有一种有效的方法来做到这一点?我知道如果我使用样本,它会混淆整个阵列。 提前致谢

6 个答案:

答案 0 :(得分:7)

一般解决方案:

编辑:正如@MatthewLundberg所评论的那样,我在seq_along(x)处理时可以轻松克服我在“x中重复数字”所指出的问题,这意味着结果值将是指数。所以,它就是这样:

k <- 3
x <- c(2,2,1, 1,3,4, 4,6,5, 3)
x.s <- seq_along(x)
y <- sample(x.s)
x[unlist(split(y, (match(y, x.s)-1) %/% k), use.names = FALSE)]
# [1] 2 2 1 3 4 1 4 5 6 3

旧答案:

这里的瓶颈是对sample的调用量。只要您的数字不重复,我认为只需拨打sample就可以这样做:

k <- 3
x <- 1:20
y <- sample(x)
unlist(split(y, (match(y,x)-1) %/% k), use.names = FALSE)
# [1]  1  3  2  5  6  4  8  9  7 12 10 11 13 14 15 17 16 18 19 20

将所有内容放在一个函数中(我喜欢@ Roland的名字scramble):

scramble <- function(x, k=3) {
    x.s <- seq_along(x)
    y.s <- sample(x.s)
    idx <- unlist(split(y.s, (match(y.s, x.s)-1) %/% k), use.names = FALSE)
    x[idx]
}

scramble(x, 3)
# [1] 2 1 2 3 4 1 5 4 6 3
scramble(x, 3)
# [1] 1 2 2 1 4 3 6 5 4 3

根据@ flodel的评论来减少答案(并加快速度):

scramble <- function(x, k=3L) {
    x.s <- seq_along(x)
    y.s <- sample(x.s)
    x[unlist(split(x.s[y.s], (y.s-1) %/% k), use.names = FALSE)]
}

答案 1 :(得分:7)

Arun不喜欢我的另一个答案是多么低效,所以这里有一些非常快的东西;)

runif()order()只需要一次通话,根本不会使用sample()

x <- 1:100
k <- 3
n <- length(x)

x[order(rep(seq_len(ceiling(n/k)), each=k, length.out=n) + runif(n))]
#  [1]   3   1   2   6   5   4   8   9   7  11  12  10  13  14  15  18  16  17
# [19]  20  19  21  23  22  24  27  25  26  29  28  30  33  31  32  36  34  35
# [37]  37  38  39  40  41  42  43  44  45  47  48  46  51  49  50  52  54  53
# [55]  55  57  56  58  60  59  62  63  61  66  64  65  68  67  69  71  70  72
# [73]  75  74  73  76  77  78  81  80  79  84  82  83  86  85  87  89  88  90
# [91]  93  92  91  94  96  95  97  98  99 100

答案 2 :(得分:5)

为了记录, boot 包(基础R附带)包含一个用于此目的的函数permutation.array()

x <- 1:100
k <- 3
ii <- boot:::permutation.array(n = length(x), 
                               R = 2, 
                               strata = (seq_along(x) - 1) %/% k)[1,]
x[ii]
#   [1]   2   1   3   6   5   4   9   7   8  12  11  10  15  13  14  16  18  17
#  [19]  21  19  20  23  22  24  26  27  25  28  29  30  33  31  32  36  35  34
#  [37]  38  39  37  41  40  42  43  44  45  46  47  48  51  50  49  53  52  54
#  [55]  57  55  56  59  60  58  63  61  62  65  66  64  67  69  68  72  71  70
#  [73]  75  73  74  76  77  78  79  80  81  82  83  84  86  87  85  89  88  90
#  [91]  93  91  92  94  95  96  97  98  99 100

答案 3 :(得分:2)

这会在最后丢弃元素(带警告):

locality <- 3
x <- 1:100
c(apply(matrix(x, nrow=locality, ncol=length(x) %/% locality), 2, sample))
## [1]  1  2  3  4  6  5  8  9  7 12 10 11 13 15 14 16 18 17 19 20 21 22 24 23 26 25 27 28 30 29 32 33 31 35 34 36 38 39 37
## [40] 42 40 41 43 44 45 47 48 46 51 49 50 54 52 53 55 57 56 58 59 60 62 61 63 64 65 66 67 69 68 71 72 70 74 75 73 78 77 76
## [79] 80 81 79 83 82 84 87 85 86 88 89 90 92 93 91 96 94 95 99 98 97

答案 4 :(得分:2)

v <- 1:16

scramble <- function(vec,n) {
  res <- tapply(vec,(seq_along(vec)+n-1)%/%n,
                FUN=function(x) x[sample.int(length(x), size=length(x))])
  unname(unlist(res))
}

set.seed(42)
scramble(v,3)
#[1]  3  2  1  6  5  4  9  7  8 12 10 11 15 13 14 16

scramble(v,4)
#[1]  2  3  1  4  5  8  6  7 10 12  9 11 14 15 16 13

答案 5 :(得分:1)

我更喜欢马修的方法,但这就是我解决问题的方式:

x <- 1:100
fact <- 3

y <- ceiling(length(x)/fact)

unlist(lapply(split(x, rep(1:y, each =fact)[1:length(x)]), function(x){
    if (length(x)==1) return(x)
    sample(x)
}), use.names = FALSE)

##   [1]   3   1   2   6   4   5   8   9   7  11  10  12  13  15  14  17  16  18
##  [19]  20  21  19  24  23  22  26  27  25  29  30  28  31  32  33  35  34  36
##  [37]  39  37  38  41  42  40  45  43  44  47  46  48  51  49  50  52  53  54
##  [55]  57  56  55  59  60  58  63  62  61  64  66  65  67  68  69  70  71  72
##  [73]  75  73  74  77  76  78  80  79  81  82  84  83  85  86  87  90  89  88
##  [91]  92  91  93  96  94  95  98  99  97 100