主页 / coding / 问题 / 77700893
Accepted
Cloft X
Asked: 2023-12-22 06:48:30 +0800 CST

如何从 R 中的给定行分配整数序列？

  • 772

我有以下数据，带有二进制变量x：

df <- data.frame(id = c("a", "a", "a", "a", "b", "b", "b", "b"),
                 year = c("2001", "2002", "2003", "2004", "2001", "2002", "2003", "2004"),
                 x = c(0, 0, 1, 0, 0, 1, 1, 0))

 id year x
  a 2001 0
  a 2002 0
  a 2003 1
  a 2004 0
  b 2001 0
  b 2002 1
  b 2003 1
  b 2004 0

我想创建一个变量，为 的第一个实例y分配整数序列。理想输出：y = 0x = 1

 id year x  y
  a 2001 0 -2
  a 2002 0 -1
  a 2003 1  0
  a 2004 0  1
  b 2001 0 -1
  b 2002 1  0
  b 2003 1  1
  b 2004 0  2

任何帮助表示赞赏！

1 个回答

  1. Best Answer
     df %>%
    mutate(y = row_number() - which(x==1)[1], .by = id)

  id year x  y
1  a 2001 0 -2
2  a 2002 0 -1
3  a 2003 1  0
4  a 2004 0  1
5  b 2001 0 -1
6  b 2002 1  0
7  b 2003 1  1
8  b 2004 0  2
    • 2

相关问题