R/helpers.R

In economiccomplexity: Computational Methods for Economic Complexity

#' country-product aggregation
#' @keywords internal
country_product_aggregation <- function(dataframe, country = "country",
                                        product = "product", value = "value") {
  dataframe <- subset(dataframe, select = c(country, product, value))
  names(dataframe) <- c("country", "product", "value")

  dataframe <- aggregate(dataframe$value, by = list(country = dataframe$country,
                         product = dataframe$product), FUN = sum)
  names(dataframe) <- c("country", "product", "value")

  dataframe <- dataframe[dataframe$value > 0, ]

  dataframe$country <- as.factor(dataframe$country)
  dataframe$product <- as.factor(dataframe$product)

  return(dataframe)
}

#' country aggregation
#' @keywords internal
country_aggregation <- function(dataframe, country = "country",
                                value = "value") {
  dataframe <- subset(dataframe, select = c(country, value))
  names(dataframe) <- c("country", "value")

  dataframe <- aggregate(dataframe$value,
                         by = list(country = dataframe$country), FUN = sum)
  names(dataframe) <- c("country", "value")

  dataframe <- dataframe[dataframe$value > 0, ]

  dataframe$country <- as.factor(dataframe$country)

  return(dataframe)
}

#' Dataframe to matrix
#' @keywords internal
dataframe_to_matrix <- function(dataframe, country = "country",
                                product = "product", value = "value") {
  countries <- as.factor(dataframe[[country]])
  products <- as.factor(dataframe[[product]])
  values <- dataframe[[value]]

  matrix_data <- matrix(0,
    nrow = length(levels(countries)), ncol = length(levels(products)),
    dimnames = list(levels(countries), levels(products))
  )

  indices <- cbind(as.numeric(countries), as.numeric(products))
  matrix_data[indices] <- values

  matrix_data
}