R/affiliationSet.R

In O1sims/networkR: A package for analysing social and economic networks

#' List the affiliation memberships of each node
#'
#' This function returns a list of the membership of each node in the hypergraph
#' @param edgeList A dataframe of network data where nodes are in the first column and affiliations are in the second column. Nodes are members of the affiliation that they are next to.
#' @param nodeList A dataframe where all nodes and their respective names are listed.
#' @keywords affiliation projection
#' @export
#' @examples
#' affiliationSet()

affiliationSet <- function(edgeList, nodeList) {
  for (i in 1:nrow(nodeList)) {
    if (i == 1) {
      affSet <- list(subset(edgeList$affiliations,
                            edgeList$nodes == nodeList[i, 1]))
    } else {
      affSet[nodeList[i, 1]] <- list(subset(edgeList$affiliations,
                                             edgeList$nodes == nodeList[i, 1]))
    }
  }
  return(affSet)
}