向向量中的随机块添加随机值

时间:2018-12-27 18:53:05

标签: r vector random

我有一个数值向量

vect <- c(0,16,11,132,0,0,0,18,28,245,0,0,55,45,19,30,20,0,0,0,12,0)

有四个不等于零的值。

(16,11,132), (18,28,245), (55,45,19,30,20), (12)

对于随机选择的1/4系列(一个系列),我想在-10和10之间添加一个随机整数值。

例如,如果所选的意甲是第二个,而所选的值是-5,则结果为

vect2 <- c(0,16,11,132,0,0,0,13,23,240,0,0,55,45,19,30,20,0,0,0,12,0)

这是仅一行的示例,该函数将应用于整个矩阵

3 个答案:

答案 0 :(得分:2)

以下不是很优雅的代码应该可以工作。它使用rle

set.seed(1)
vect <- c(0,16,11,132,0,0,0,18,28,245,0,0,55,45,19,30,20,0,0,0,12,0)

# Get non-zero runs
runs <- rle(vect!=0)
non_zero_runs <- with(runs, lengths[values])

# Sample the required
rand_run <- sample(seq_along(non_zero_runs), size = 1)
print(rand_run)
#[1] 2
rand_int <- sample(-10:10, size = 1)
print(rand_int)
#[1] -3

# Identify the sampled run
which_run <- min(which(cumsum(runs$values) == rand_run))

# Identify the corresponding indices
start <- sum(runs$lengths[seq_len(which_run - 1)]) + 1
end <- start + runs$lengths[which_run] - 1

vect[start:end] <- vect[start:end] + rand_int
print(vect)
#[1]   0  16  11 132   0   0   0  15  25 242   0   0  55  45  19  30  20   0   0   0  12   0

您可以尝试自己减少代码。该代码非常冗长,因为它是解决和理解该问题的一部分。

答案 1 :(得分:2)

不使用rle的解决方案。在这里,我使用diff查找不是0的职位。然后随机选择一个位置,并生成除一个位置外的零向量。

# Extract positions that are not 0
sequences <- c(0, diff(vect != 0))

# Get starts
starts <- which(sequences == 1)
# One random position from starts
position <- sample(seq_along(starts), 1)

# Get absolute start and end positions
currentStart <- starts[position]
currentEnd <- which(sequences == -1)[position] - 1

# Add this vector to originak
addVector <- rep(0, length(vect))
addVector[currentStart:currentEnd] <- rep(sample(-10:10, 1), currentEnd - currentStart + 1)

vect + addVector

答案 2 :(得分:1)

这里是一线人:

vect + with(rle(vect != 0), rep(replace(0 * values, which(values)[sample(sum(values), 1)], sample(-10:10, 1)), lengths))
# [1]   0  16  11 132   0   0   0  18  28 245   0   0  55  45  19  30  20   0   0   0   8   0

更详细的说明

tmp <- rle(vect != 0)
add <- sample(-10:10, 1)
tmp$values <- replace(0 * tmp$val, which(tmp$val)[sample(sum(tmp$val), 1)], add)
vect + inverse.rle(tmp)

因此,我们通过将第二个分量添加到初始vect中来构造一个新向量。部分

replace(0 * tmp$val, which(tmp$val)[sample(sum(tmp$val), 1)], add)
# [1] 0 3 0 0 0 0 0 0 0

选择一个非零块,然后在其中放置一个新的随机值add。然后

inverse.rle(tmp)
# [1] 0 3 3 3 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0