Пропуск NA строк в столбце индекса

Question

У меня есть фрейм данных со столбцом со значениями TRUE и NA. Я хотел бы добавить столбец индекса к df, который индексирует все строки с TRUE, но просто пропускает все строки с NA, не удаляя их. Вот так:

VarA    ID
TRUE    1
NA      NA
NA      NA
TRUE    2
TRUE    3

На данный момент я делаю это как:

i = 1:nrow(df)
df$ID <- ifelse(df$VarA == TRUE, i, NA)

, но это дает мне

VarA    ID
TRUE    1
NA      NA
NA      NA
TRUE    4
TRUE    5

, а это не то, что я хочу , Я знаю, что это должно быть очень легко с чем-то вроде na.omit, но почему-то я просто не могу понять это.

Answer 1

Использование базы R

df1$ID2[!is.na(df1$VarA)] <- seq_along(na.omit(df1$VarA))

  VarA ID ID2
1 TRUE  1   1
2   NA NA  NA
3   NA NA  NA
4 TRUE  2   2
5 TRUE  3   3

Answer 2

# example dataset
df = data.frame(VarA = c(TRUE, NA, NA, TRUE, TRUE))

# get cumulative sum based on cases where you don't have NA in VarA
df$ID = cumsum(!is.na(df$VarA))

# if you have NA in VarA use NA instead ofthe cumulative sum
df$ID = ifelse(is.na(df$VarA), df$VarA, df$ID)

df

#   VarA ID
# 1 TRUE  1
# 2   NA NA
# 3   NA NA
# 4 TRUE  2
# 5 TRUE  3

Вы можете объединить вышеперечисленное в одной команде, если вы предпочитаете, как

df$ID = ifelse(is.na(df$VarA), df$VarA, cumsum(!is.na(df$VarA)))

Answer 3

Вы можете использовать индекс i, содержащий !is.na(df$VarA), который можно использовать для подстановки df и для генерации последовательности.

i <- !is.na(df$VarA)
df$ID[i] <- seq_len(sum(i))
df
#  VarA ID
#1 TRUE  1
#2   NA NA
#3   NA NA
#4 TRUE  2
#5 TRUE  3

Или используйте which для индекса:

i <- which(df$VarA)
df$ID[i] <- seq_along(i)

Данные:

df <- data.frame(VarA = c(TRUE,NA,NA,TRUE,TRUE))

Answer 4

Базовый раствор R с использованием which

df$ID[which(df$VarA)]<-seq(which(df$VarA))

Answer 5

data.table решение

library( data.table )
DT <- data.table::fread("
VarA
TRUE
NA
NA
TRUE
TRUE")

DT[ !is.na(VarA), ID := .I][]

или

setDT(df)[ !is.na(VarA), ID := .I][]

приводит к

#    VarA ID
# 1: TRUE  1
# 2:   NA NA
# 3:   NA NA
# 4: TRUE  2
# 5: TRUE  3

Answer 6

Один вариант в base:

transform(df, ID = replace(cumsum(!is.na(VarA)), is.na(VarA), NA))

Выход:

   VarA ID
1: TRUE  1
2:   NA NA
3:   NA NA
4: TRUE  2
5: TRUE  3

Answer 7

Используя tidyverse, вы можете сделать это:

library(tidyverse)

df %>% 
  group_by(VarA) %>%
  mutate(counter = row_number(ID))

Вывод:

  VarA     ID counter
  <lgl> <int>   <int>
1 TRUE      1       1
2 NA       NA      NA
3 NA       NA      NA
4 TRUE      2       2
5 TRUE      3       3

Пример данных:

df <- read.table(text = "VarA    ID
TRUE    1
           NA      NA
           NA      NA
           TRUE    2
           TRUE    3", header = TRUE,
           stringsAsFactors = FALSE)

Answer 8

Нет необходимости в цикле или итерации:

df$ID <- ifelse(df$VarA == TRUE, TRUE, NA)

Должен дать желаемый результат.

Или, может быть, я неправильно понял, и вы хотите удалить строк с NA in VarA:

df[!is.na(df$VarA), ]