5

我想将我的数据分成通过一些测试的连续行组。这是一个例子：

set.seed(1)
n   <-  29
ok  <-  sample(c(TRUE,FALSE),n,replace=TRUE,prob=c(.7,.3))

vec <- (1:n)[ok]
# [1]  1  2  3  5  8  9 10 11 12 13 14 16 19 22 23 24 25 26 27 28

所需的输出是“vec”，分组为连续序列：

out <- list(1:3,5,8:14,16,19,22:28)

这有效：

nv  <- length(vec)

splits <- 1 + which(diff(vec) != 1)
splits <- c(1,splits,nv+1)
nsp    <- length(splits)

out <- list()
for (i in 1:(nsp-1)){
    out[[i]] <- vec[splits[i]:(splits[i+1]-1)]
}

我猜在基础 R 中有一种更清洁的方法......？我还不擅长我在 SO 上看到的rle和技巧......cumsum

4

1 回答 1

6

这里有一个cumsum“技巧”给你：

split(vec, cumsum(c(1, diff(vec)) - 1))

更新

这是一个使用您的版本的简单示例split(vec, cumsum(c(0, diff(vec) > 1)))，每个步骤都被分解：

vec <- c(1:3,7:9)            #  1 2 3 7 8 9 (sample with two contiguous sequences)
diff(vec)                    #   1 1 4 1 1  (lagged difference)
diff(vec) > 1                #   F F T F F  (not contiguous where diff > 1)
                             #   0 0 1 0 0  (numeric equivalent for T/F)
c(0, diff(vec) > 1)          #  0 0 0 1 0 0 (pad with 0 to align with original vector)
cumsum(c(0, diff(vec) > 1))  #  0 0 0 1 1 1 (cumulative sum of logical values)

groups <- cumsum(c(0, diff(vec) > 1)) #  0 0 0 1 1 1

sets <- split(vec, groups) # split into groups named by cumulative sum

sets
# $`0`
# [1] 1 2 3
# 
# $`1`
# [1] 7 8 9

然后，如果您出于某种原因想要输出它：

# Create strings representing each contiguous range
set_strings <- sapply(sets, function(x) paste0(min(x),":",max(x)))

set_strings
#     0     1
# "1:3" "7:9"

# Print out a concise representation of all contiguous sequences
print(paste0(set_strings,collapse=","))

# [1] "1:3,7:9"
于 2013-05-28T20:29:37.773 回答