Início / coding / Perguntas / 77700893
Accepted
Cloft X
Asked: 2023-12-22 06:48:30 +0800 CST

Como atribuir uma sequência de inteiros de uma determinada linha em R?

  • 772

Tenho os seguintes dados, com variável binária 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

Quero criar uma variável yque atribua uma sequência de números inteiros y = 0para a primeira instância de x = 1. A saída ideal:

 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

Qualquer ajuda é apreciada!

1 respostas

  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

relate perguntas