生成逆置换

时间:2013-01-03 23:39:17

标签: r vector permutation

假设我们有一个向量foo,我们必须临时置换它(排序或重新排序),在它的基础上计算一些向量bar ,最后将foobar置换为foo的原始顺序 - 这意味着反向排列:

foo <- c(1, 7, 3, 5, 2)
o <- order(foo)
foo <- foo[o] # Now foo is permuted, and sorted: foo == 1 2 3 5 7
bar = 2 * foo # bar == 2 4 6 10 14

这里应该给你答案,以便我们得到以下所需的最终值:

foo == 1 7 3 5 2
bar == 2 14 6 10 4

怎么做?

请简单地不回复:“你可以做bar = 2 * foo而不是置换它”。这只是一个简单的例子。在某些情况下,我们必须对foo进行排序以提高效率(快速搜索它)或类似的东西。

2 个答案:

答案 0 :(得分:13)

这样可行,因为[order(o)]会颠倒[o]

的操作
foo <- c(1, 7, 3, 5, 2)
o <- order(foo)

(foo[o]*2)[order(o)]
# [1]  2 14  6 10  4

表明它更常用:

testAlgorithm <- function(foo) {
    o <- order(foo)
    identical(foo, foo[o][order(o)])
}

x <- c(1, 7, 3, 5, 2)
y <- c(1,2,5,7,4, 99, 88, 3, 0)
z <- sample(1000)                  ## (No ties)
zz <- sample(1000, replace=TRUE)   ## (Many ties)
all(sapply(list(x,y,z,zz), testAlgorithm))
[1] TRUE

答案 1 :(得分:8)

要恢复原始订单,请使用order(o)

> (foo[o]*2)[order(o)]
[1]  2 14  6 10  4