R 트리맵 - treemapify 로 그래프 그리기

아래 그래프같이 R 프로그래밍으로도 트리맵을 그릴 수 있다. ggplot2 라이브러리만으로는 안되고 treemapify 라는 라이브러리를 설치를 해야 한다. http://wilkox.org/treemapify/ 에서 설치방법은 참고하시기 바란다. 

 

트리맵 그래프에 쓰일 데이터를 한번 만들어본다. 트리맵 구조상 값이 비슷비슷하면 모양이 이쁘지 않고 비교가 쉽지 않기 때문에, 네제곱해서 값간의 간극을 다이나믹하게 만들어본다.

set.seed(392890)
tb1 = tibble(x = LETTERS[1:9], 
             y = sort(cumsum(sample(50:300, 9)), decreasing = T)^4/1000000)
tb1
# A tibble: 9 × 2
  x            y
  <chr>    <dbl>
1 A     9770776.
2 B     5116716.
3 C     3282315.
4 D     1584789.
5 E      856447.
6 F      333622.
7 G       51769.
8 H       27710.
9 I         672.

 

 

데이터가 준비되었다면 2줄의 코딩으로 트리맵을 간단히 만들 수 있다.

library(treemapify)
ggplot(tb1, aes(area = y, fill = y)) +
  geom_treemap()

 

이제는 텍스트를 추가해볼 차례다. geom_treemap_text 를 추가하기만 하면 된다.

ggplot(tb1, aes(area = y, fill = x)) +
  geom_treemap() +
  geom_treemap_text(aes(label = x))

반응형

텍스트를 하나더 추가한다. 그러기 위해서 기존 데이터셋에 비중을 구해놓아야 하는데, 아래는 예시일 뿐이니 참고하면 된다. 45% 같은 텍스트를 그냥 추가하면, 기존 A~I 로 표시된 위치에 덮어 써지기 때문에 위치를 조금 달리해야 한다. place 라는 옵션을 사용하면 되고, bottom, top, left, right 를 잘 조합해서 쓰면 원하는 위치에 텍스트를 표시할 수 있다.

set.seed(392890)
tb2 = tibble(x = LETTERS[1:9], 
             y = sort(cumsum(sample(50:300, 9)), decreasing = T)^4/1000000) %>% 
  mutate(l = paste0(round(y / sum(y) * 100, 1), "%"))
  
ggplot(tb2, aes(area = y, fill = x)) +
  geom_treemap() +
  geom_treemap_text(aes(label = x)) +
  geom_treemap_text(aes(label = l), place = "bottomright")

 

 

그리고, 가장 높은 값(비중)이 지금은 좌측하단부터 시작하고 있는데, 좌측상단부터 시작되도록 변경해본다. start 옵션에 "topleft" 를 지정하기만 하면된다.

ggplot(tb2, aes(area = y, fill = x)) +
  geom_treemap(start = "topleft") +
  geom_treemap_text(aes(label = x), start = "topleft") +
  geom_treemap_text(aes(label = l), start = "topleft", place = "bottomright")

 

 

트리맵 중간중간에 간격을 주고 싶으면, colour 를 흰색으로 설정하고 size 를 지정하면 된다. 아래 예시로 참고 하면 된다.

ggplot(tb2, aes(area = y, fill = x)) +
  geom_treemap(start = "topleft", colour = "gray100", size = 5) +
  geom_treemap_text(aes(label = x), start = "topleft", colour = "gray100") +
  geom_treemap_text(aes(label = l), start = "topleft", place = "bottomright", colour = "gray100") +
  scale_fill_brewer(direction = -1)