import-export-states-sankey

sankey-code.R

library(networkD3)
library(magrittr)
library(dplyr)
library(readr)
library(tidyr)

tmp <- read_csv('sankey-data.csv', skip = 2)
tmp <- tmp[-(21:22),]
names(tmp) <- c("X1", "Maharashtra", "Gujarat", "Karnataka", "Andhra Pradesh", 
                "Uttarakhand", "Rajasthan", "Madhya Pradesh", "West Bengal", "Tamil Nadu", 
                "Kerala", "Odisha", "Uttar Pradesh", "Delhi", "Chattisgarh", 
                "Himachal Pradesh", "Haryana", "Bihar", "Jharkhand", "Goa", "Assam")
tmp$X1 <- c("Maharashtra", "Gujarat", "Tamil Nadu", "Haryana", "Karnataka", 
            "Andhra Pradesh", "Uttar Pradesh", "Rajasthan", "Delhi", "West Bengal", 
            "Madhya Pradesh", "Uttarakhand", "Odisha", "Chattisgarh", "Jharkhand", 
            "Himachal Pradesh", "Kerala", "Goa", "Assam", "Bihar")

tmp <- tmp[, c("X1", "Maharashtra", "Gujarat", "Tamil Nadu", "Haryana", "Karnataka", 
               "Andhra Pradesh", "Uttar Pradesh", "Rajasthan", "Delhi", "West Bengal", 
               "Madhya Pradesh", "Uttarakhand", "Odisha", "Chattisgarh", "Jharkhand", 
               "Himachal Pradesh", "Kerala", "Goa", "Assam", "Bihar")]

tmp <- tmp %>% 
  gather(key = 'import_state', value = 'value', -X1) %>% 
  rename(export_state = X1) %>% 
  mutate(value = as.numeric(value)) %>% 
  filter(!is.na(value))

trade_surplus <- bind_rows(
  tmp %>% 
    group_by(export_state) %>% 
    summarise(value = sum(value)) %>% 
    ungroup %>% 
    rename(state = export_state),
  tmp %>% 
    group_by(import_state) %>% 
    summarise(value = -sum(value)) %>% 
    ungroup %>% 
    rename(state = import_state)
) %>% 
  group_by(state) %>% 
  summarise(value = sum(value)) %>% 
  ungroup

export_map <- tmp$export_state %>% unique %>% seq_along() - 1
names(export_map) <- tmp$export_state %>% unique %>% sort

import_map <- tmp$import_state %>% unique %>% seq_along() + 19
names(import_map) <- tmp$import_state %>% unique %>% sort

tmp2 <- list(nodes = data.frame(name = c(
  tmp$export_state %>% unique %>% sort,
  tmp$import_state %>% unique %>% sort
)) %>% mutate(name = as.character(name))
)

tmp2$links <- tmp %>%
  transmute(
    source = export_map[export_state] %>% unname,
    target = import_map[import_state] %>% unname,
    value = as.numeric(value),
    export_state = export_state
  ) %>% data.frame

gg <- sankeyNetwork(Links = tmp2$links, Nodes = tmp2$nodes, Source = "source",
                    Target = "target", Value = "value", NodeID = "name",
                    units = "", fontSize = 12, nodeWidth = 30, iterations = 100,
                    LinkGroup = "export_state")

gg

saveNetwork(gg, 'tmp.html', selfcontained = TRUE)

sankey-data.csv