计算R中相同符号的累积值和顺序值

Question

我正在寻找 R 中与 python 中this post 的等效代码，用于添加一个列，该列累积计算前一列中正值和负值的数量。

我发现了许多累积和或更复杂的示例，但我只想计算一排中正数和负数的数量，只要符号发生变化就会重置。请参阅示例代码。

library(dplyr)
df <- data.frame(x = c(0.5, 1, 6.5, -2, 3, -0.2, -1))

我的预期输出是这样的：

df <- data.frame(x = c(0.5, 1, 6.5, -2, 3, -0.2, -1), 
                 z = c(1,2,3,-1,1,-1,-2))

当数据帧 df 仅以“x”开头时，我希望 R 使用 mutate 函数创建列“z”。

Answer 1

你可以试试：

library(dplyr)

df %>%
  mutate(z = with(rle(sign(x)), sequence(lengths) * rep(values, lengths)))

     x  z
1  0.5  1
2  1.0  2
3  6.5  3
4 -2.0 -1
5  3.0  1
6 -0.2 -1
7 -1.0 -2

您可能需要考虑如何处理零，因为如果您的向量中存在零，上述可能需要修改。也许：

df %>%
   mutate(z = with(rle(sign(x)), sequence(lengths) * rep(values^(values != 0), lengths)))

编辑下面的寻址 OP 评论：

df %>%
  mutate(z = with(tmp <- rle(sign(x)), sequence(lengths) * rep(values, lengths)),
         id = with(tmp, rep(seq_along(lengths), lengths))) %>%
  group_by(id) %>%
  mutate(avg = cumsum(x)/row_number()) %>%
  ungroup() %>%
  select(-id)

# A tibble: 7 x 3
      x     z   avg
  <dbl> <dbl> <dbl>
1   0.5     1  0.5 
2   1       2  0.75
3   6.5     3  2.67
4  -2      -1 -2   
5   3       1  3   
6  -0.2    -1 -0.2 
7  -1      -2 -0.6