Question

我有以下问题，我找不到任何简单的解决方案。

我想在向量中找到非零元素组（由至少一个零分隔）并为每个组分配一个id（后续整数）。例如，在向量value <- c(1,1,2,3,4,3,0,0,0,1,2,3,9,8,0,0,3,2)中应该有三组：[1,1,2,3,4,3]，[1,2,3,9,8]，[3,2]，所以我我想获得这样一个数据框：

   value id
1      1  1
2      1  1
3      2  1
4      3  1
5      4  1
6      3  1
7      0 NA
8      0 NA
9      0 NA
10     1  2
11     2  2
12     3  2
13     9  2
14     8  2
15     0 NA
16     0 NA
17     3  3
18     2  3

Answer 1

您可以尝试：

as.integer(factor(cumsum(value==0)*NA^(value==0)))
#[1]  1  1  1  1  1  1 NA NA NA  2  2  2  2  2 NA NA  3  3

Answer 2

使用rle()。首先创建一个用NA替换零的新向量。

x <- match(value != 0, TRUE)
with(rle(!is.na(x)), {
    lv <- lengths[values]
    replace(x, !is.na(x), rep(seq_along(lv), lv))
})
# [1]  1  1  1  1  1  1 NA NA NA  2  2  2  2  2 NA NA  3  3

Answer 3

你也可以这样做：

id <- (value>0)^NA
x <- rle(value>0)$lengths[c(TRUE, FALSE)]
id[!is.na(id)] <- rep(seq_along(x), times=x)

#[1]  1  1  1  1  1  1 NA NA NA  2  2  2  2  2 NA NA  3  3

Answer 4

另一种可能性：

ifelse(value != 0, 
       cumsum(value != 0 & dplyr::lag(value) %in% c(0, NA)), 
       NA)

# [1]  1  1  1  1  1  1 NA NA NA  2  2  2  2  2 NA NA  3  3

Answer 5

您需要定义矢量矢量，因此在v [0]中您将找到第一组的所有值，在v [1]中您将找到第二组的所有值，依此类推，
当您找到零值时，需要循环所有值。继续，直到你找到零。使用一个增加向量并添加值，依此类推，

我希望这个答案有所帮助。

使用R查找非零元素组

5 个答案: