R/getEdges.R

In netknitr: Knit Network Map for any Dataset

#' getEdges
#'
#' @param polished_data data frame, output of getAssociation function
#' @param nodes data frame, output of getNodes function
#' @author Jayachandra N
#' @description
#' Generate edges or lines data frame which defines the link between nodes.
#'
#' @return data frame of edges indicationg from and to node ids
#' @export
#' @examples
#' res <- fixNodeBias(head(mtcars))
#' edges <- getEdges(getAssociation(res), getNodes(res, group = TRUE))
getEdges <- function(polished_data, nodes) {
  from <- lapply(seq_len(nrow(polished_data)), FUN = function(x) {
    pos <- which(as.character(nodes$label) == as.character(polished_data$from[x]))
    nodes[pos, ]$id
  })
  from <- do.call("c", from)
  to <- lapply(seq_len(nrow(polished_data)), FUN = function(x) {
    pos <- which(as.character(nodes$label) == as.character(polished_data$to[x]))
    nodes[pos, ]$id
  })
  to <- do.call("c", to)
  edges <- data.frame(from = from, to = to)
  edges
}