Как я могу сбалансировать панель, которая не сбалансирована в R? - PullRequest
Новая
       26

Как я могу сбалансировать панель, которая не сбалансирована в R?

0 голосов
/ 24 января 2019

У меня несбалансированные месячные данные панели. В некоторых штатах отсутствуют некоторые периоды времени, и поэтому у меня нет этих периодов в моей группе. Я хотел бы добавить эти периоды и присвоить им значение NA.

Я пробовал функцию, которая балансирует панель, но она устраняет все наблюдения, которые не соответствуют всем состояниям. Поэтому, если нет данных по Алабаме за 02-2008, это исключило бы наблюдения для всех штатов за 02-2008. 

balanced <- function(data, ID, TIME, VARS, required=c("all", "shared")) {
  if(is.character(ID)) {
    ID <- match(ID, names(data))
  }
  if(is.character(TIME)) {
    TIME <- match(TIME, names(data))
    if(missing(VARS)) { 
      VARS <- setdiff(1:ncol(data), c(ID,TIME))
    } else if (is.character(VARS)) {
      VARS <- match(VARS, names(data))
    }
    required <- match.arg(required)
    idf <- do.call(interaction, c(data[, ID, drop=FALSE], drop=TRUE))
    timef <- do.call(interaction, c(data[, TIME, drop=FALSE], drop=TRUE))
    complete <- complete.cases(data[, VARS])
    tbl <- table(idf[complete], timef[complete])
    if (required == "all") {
      keep <- which(rowSums(tbl == 1) == ncol(tbl))
      idx <- as.numeric(idf) %in% keep
    } else if (required == "shared") {
      keep <- which(colSums(tbl == 1) == nrow(tbl))
      idx <- as.numeric(timef) %in% keep
    }
    data[idx, ]
  }

Balanced_panel <- balanced(milk_state, "STATE", "PERIOD", "VALUE", required = "all")

Вот как это выглядит: 

  STATE       PERIOD    VALUE
California    01-2018     900
California    02-2018     890
California    03-2018     780
California    05-2018     800

Вот как я хочу, чтобы это выглядело: 

  STATE       PERIOD    VALUE
California    01-2018     900
California    02-2018     890
California    03-2018     780
California    04-2018      NA
California    05-2018     800
...