[R語言]數(shù)據(jù)可視化的最佳解決方案:ggplot2

前言

ggplot是一個擁有一套完備語法且容易上手的繪圖系統(tǒng),在PythonR中都能引入并使用,在數(shù)據(jù)分析可視化領(lǐng)域擁有極為廣泛的應(yīng)用。本篇從R的角度介紹如何使用ggplot2包,首先給幾個我覺得最值得推薦的理由:

  • 采用“圖層”疊加的設(shè)計方式,一方面可以增加不同的圖之間的聯(lián)系,另一方面也有利于學(xué)習(xí)和理解該packagephotoshop的老玩家應(yīng)該比較能理解這個帶來的巨大便利
  • 適用范圍廣,擁有詳盡的文檔,通過?和對應(yīng)的函數(shù)即可在R中找到函數(shù)說明文檔和對應(yīng)的實例
  • RPython中均可使用,降低兩門語言之間互相過度的學(xué)習(xí)成本

基本概念

本文采用ggplot2的自帶數(shù)據(jù)集diamonds。

> head(diamonds)
# A tibble: 6 x 10
  carat cut       color clarity depth table price     x     y     z
  <dbl> <ord>     <ord> <ord>   <dbl> <dbl> <int> <dbl> <dbl> <dbl>
1 0.23  Ideal     E     SI2      61.5    55   326  3.95  3.98  2.43
2 0.21  Premium   E     SI1      59.8    61   326  3.89  3.84  2.31
3 0.23  Good      E     VS1      56.9    65   327  4.05  4.07  2.31
4 0.290 Premium   I     VS2      62.4    58   334  4.2   4.23  2.63
5 0.31  Good      J     SI2      63.3    58   335  4.34  4.35  2.75
6 0.24  Very Good J     VVS2     62.8    57   336  3.94  3.96  2.48

# 變量含義
price  : price in US dollars (\$326–\$18,823)
carat  : weight of the diamond (0.2–5.01)
cut    :   quality of the cut (Fair, Good, Very Good, Premium, Ideal)
color  : diamond colour, from D (best) to J (worst)
clarity: a measurement of how clear the diamond is (I1 (worst), SI2, SI1, VS2, VS1, VVS2, VVS1, IF (best))
x      : length in mm (0–10.74)
y      : width in mm (0–58.9)
z      : depth in mm (0–31.8)
depth  : total depth percentage = z / mean(x, y) = 2 * z / (x + y) (43–79)
table  : width of top of diamond relative to widest point (43–95)

基于圖層和畫布的概念,ggplot2引申出如下的語法框架:

圖源:https://mp.weixin.qq.com/s/uskZWGAwfK9BVqLBQIXpGA

image.png
  • data:數(shù)據(jù)源,一般是data.frame結(jié)構(gòu),否則會被轉(zhuǎn)化為該結(jié)構(gòu)
  • 個性映射與共性映射:ggplot()中的mapping = aes()參數(shù)屬于共性映射,會被之后的geom_xxx()stat_xxx()所繼承,而geom_xxx()stat_xxx()中的映射參數(shù)屬于個性映射,僅作用于內(nèi)部
  • mapping:映射,包括顏色類型映射color;fill、形狀類型映射linetype;size;shape和位置類型映射x,y
  • geom_xxx:幾何對象,常見的包括點圖、折線圖、柱形圖和直方圖等,也包括輔助繪制的曲線、斜線、水平線、豎線和文本等
  • aesthetic attributes:圖形參數(shù),包括colour;size;hape
  • facetting:分面,將數(shù)據(jù)集劃分為多個子集subset,然后對于每個子集都繪制相同的圖表
  • theme:指定圖表的主題
ggplot(data = NALL, mapping = aes(x = , y = )) +   # 數(shù)據(jù)集
    geom_xxx()|stat_xxx() +       # 幾何圖層/統(tǒng)計變換
    coord_xxx() +  # 坐標變換, 默認笛卡爾坐標系     
    scale_xxx() +  # 標度調(diào)整, 調(diào)整具體的標度  
    facet_xxx() +  # 分面, 將其中一個變量進行分面變換  
    guides() +     # 圖例調(diào)整
    theme()        # 主題系統(tǒng)

這些概念可以等看完全文再回過頭看,相當(dāng)于一個匯總,這些概念都掌握了基本ggplot2的核心邏輯也就理解了

一些核心概念的含義可以從RStudio官方的cheat sheet圖中大致得知:

image.png

image.png

一些栗子

通過實例和RCode從淺到深介紹ggplot2的語法。

1. 五臟俱全的散點圖

library(ggplot2)

# 表明我們使用diamonds數(shù)據(jù)集, 
ggplot(diamonds) + 
  # 繪制散點圖: 橫坐標x為depth, 縱坐標y為price, 點的顏色通過color列區(qū)分,alpha透明度,size點大小,shape形狀(實心正方形),stroke點邊框的寬度
  geom_point(aes(x = carat, y = price, colour = color), alpha=0.7, size=1.0, shape=15, stroke=1) +
  # 添加擬合線
  geom_smooth(aes(x = carat, y = price), method = 'glm') +
  # 添加水平線
  geom_hline(yintercept = 0, size = 1, linetype = "dotted", color = "black") +
  # 添加垂直線
  geom_vline(xintercept = 3, size = 1, linetype = "dotted", color = "black") +
  # 添加坐標軸與圖像標題
  labs(title = "Diamonds Point Plot", x = "Carat", y = "Price") +
  # 調(diào)整坐標軸的顯示范圍
  coord_cartesian(xlim = c(0, 3), ylim = c(0, 20000)) +
  # 更換主題, 這個主題比較簡潔, 也可以在ggthemes包中獲取其他主題
  theme_linedraw()
image.png

2. 自定義圖片布局&多種幾何繪圖

library(gridExtra)
#建立數(shù)據(jù)集
df <- data.frame(
  x = c(3, 1, 5),
  y = c(2, 4, 6),
  label = c("a","b","c")
)  

p <- ggplot(df, aes(x, y, label = label)) +
  # 去掉橫坐標信息
  labs(x = NULL, y = NULL) +
  # 切換主題
  theme_linedraw()

p1 <- p + geom_point() + ggtitle("point")
p2 <- p + geom_text() + ggtitle("text")
p3 <- p + geom_bar(stat = "identity") + ggtitle("bar")
p4 <- p + geom_tile() + ggtitle("raster")
p5 <- p + geom_line() + ggtitle("line")
p6 <- p + geom_area() + ggtitle("area")
p7 <- p + geom_path() + ggtitle("path")
p8 <- p + geom_polygon() + ggtitle("polygon")

# 構(gòu)造ggplot圖片列表
plots <- list(p1, p2, p3, p4, p5, p6, p7, p8)
# 自定義圖片布局
gridExtra::grid.arrange(grobs = plots, ncol = 4)
image.png

3. 箱線圖

統(tǒng)計學(xué)中展示數(shù)據(jù)分散情況的直觀圖形,在探索性分析中常常用于展示在某個因子型變量下因變量的分散程度。

下面展示箱線圖最長使用的一些方法:

library(ggplot2) # 繪圖
library(ggsci)   # 使用配色

# 使用diamonds數(shù)據(jù)框, 分類變量為cut, 目標變量為depth
p <- ggplot(diamonds, aes(x = cut, y = carat)) +
  theme_linedraw()

# 一個因子型變量時, 直接用顏色區(qū)分不同類別, 后面表示將圖例設(shè)置在右上角
p1 <- p + geom_boxplot(aes(fill = cut)) + theme(legend.position = "None")
# 兩個因子型變量時, 可以將其中一個因子型變量設(shè)為x, 將另一個因子型變量設(shè)為用圖例顏色區(qū)分
p2 <- p + geom_boxplot(aes(fill = color)) + theme(legend.position = "None")
# 將箱線圖進行轉(zhuǎn)置
p3 <- p + geom_boxplot(aes(fill = cut)) + coord_flip() + theme(legend.position = "None")
# 使用現(xiàn)成的配色方案: 包括scale_fill_jama(), scale_fill_nejm(), scale_fill_lancet(), scale_fill_brewer()(藍色系)
p4 <- p + geom_boxplot(aes(fill = cut)) + scale_fill_brewer() + theme(legend.position = "None")

# 構(gòu)造ggplot圖片列表
plots <- list(p1, p2, p3, p4)
# 自定義圖片布局
gridExtra::grid.arrange(grobs = plots, ncol = 2)
image.png

當(dāng)研究某個連續(xù)型變量的箱線圖涉及多個離散型分類變量時,我們常使用分面facetting來提高圖表的可視性。

library(ggplot2)

ggplot(diamonds, aes(x = color, y = carat)) +
  # 切換主題
  theme_linedraw() +
  # 箱線圖顏色根據(jù)因子型變量color填色
  geom_boxplot(aes(fill = color)) +
  # 分面: 本質(zhì)上是將數(shù)據(jù)框按照因子型變量color類劃分為多個子數(shù)據(jù)集subset, 在每個子數(shù)據(jù)集上繪制相同的箱線圖
  # 注意一般都要加scales="free", 否則子數(shù)據(jù)集數(shù)據(jù)尺度相差較大時會被拉扯開
  facet_wrap(~cut, scales="free")
image.png

4. 直方圖

library(ggplo2)

# 普通的直方圖
p1 <- ggplot(data = diamonds) + 
  geom_bar(mapping = aes(x = cut, fill = cut)) + 
  theme_linedraw() +
  scale_fill_brewer()

# 堆積直方圖
p2 <- ggplot(data = diamonds) + 
  geom_bar(mapping = aes(x = cut, fill = clarity), position = "identity") + 
  theme_linedraw() +
  scale_fill_brewer()
  
# 累積直方圖
p3 <- ggplot(data = diamonds) + 
  geom_bar(mapping = aes(x = cut, fill = clarity), position = "fill") + 
  theme_linedraw() +
  scale_fill_brewer()

# 分類直方圖
p4 <- ggplot(data = diamonds) + 
  geom_bar(mapping = aes(x = cut, fill = clarity), position = "dodge") + 
  theme_linedraw() +
  scale_fill_brewer()

# 構(gòu)造ggplot圖片列表
plots <- list(p1, p2, p3, p4)
# 自定義圖片布局
gridExtra::grid.arrange(grobs = plots, ncol = 2)
image.png

5. 坐標系統(tǒng)

除了前面箱線圖使用的coord_flip()方法實現(xiàn)了坐標軸轉(zhuǎn)置,ggplot還提供了很多和坐標系統(tǒng)相關(guān)的功能。

library(ggplot2)

bar <- ggplot(data = diamonds) + 
  geom_bar(mapping = aes(x = cut, fill = cut), show.legend = FALSE, width = 1) + 
  # 指定比率: 長寬比為1, 便于展示圖形
  theme(aspect.ratio = 1) +
  scale_fill_brewer() +
  labs(x = NULL, y = NULL)

# 坐標軸轉(zhuǎn)置
bar1 <- bar + coord_flip()
# 繪制極坐標
bar2 <- bar + coord_polar()

# 構(gòu)造ggplot圖片列表
plots <- list(bar1, bar2)
# 自定義圖片布局
gridExtra::grid.arrange(grobs = plots, ncol = 2)
image.png

6. 瓦片圖、 熱力圖

機器學(xué)習(xí)中探索性分析我們可以通過corrplot直接繪制所有變量的相關(guān)系數(shù)圖,用于判斷總體的相關(guān)系數(shù)情況。

library(corrplot)
#計算數(shù)據(jù)集的相關(guān)系數(shù)矩陣并可視化
mycor = cor(mtcars)
corrplot(mycor, tl.col = "black")
image.png

ggplot提供了更加個性化的瓦片圖繪制:

library(RColorBrewer)
# 生成相關(guān)系數(shù)矩陣
corr <- round(cor(mtcars), 2)
df <- reshape2::melt(corr)
p1 <- ggplot(df, aes(x = Var1, y = Var2, fill = value, label = value)) +
  geom_tile() +
  theme_bw() +
  geom_text(aes(label = value, size = 0.3), color = "white") +
  labs(title = "mtcars - Correlation plot") +
  theme(text = element_text(size = 10), legend.position = "none", aspect.ratio = 1)
p2 <- p1 + scale_fill_distiller(palette = "Reds")
p3 <- p1 + scale_fill_gradient2()
gridExtra::grid.arrange(p1, p2, p3, ncol=3)
image.png

更多例子

有經(jīng)典的50個ggplot2繪圖示例:

http://r-statistics.co/Top50-Ggplot2-Visualizations-MasterList-R-Code.html

其他文章

1. 機器學(xué)習(xí)必知必會與算法原理

機器學(xué)習(xí)導(dǎo)論:什么是機器學(xué)習(xí)
機器學(xué)習(xí)必知必會:凸優(yōu)化
深入淺出機器學(xué)習(xí)算法:XGBoost
機器學(xué)習(xí)必知必會:梯度下降法

2. 數(shù)據(jù)分析和爬蟲案例

Python數(shù)據(jù)分析:誰是2018當(dāng)之無愧的“第一”國產(chǎn)電影
如何用python爬蟲實現(xiàn)簡單PV刷量——以CSDN為例
python腳本從零到一構(gòu)建自己的免費代理IP池

3. 相關(guān)經(jīng)驗

秋招面試:零基礎(chǔ)拿到騰訊數(shù)據(jù)崗offer需要做哪些努力
股票市場中如何用數(shù)據(jù)思維跑贏九成的投資者
精算師證有多難考,怎么準備?

Reference

[1] https://ggplot2-book.org/introduction.html#welcome-to-ggplot2
[2] https://rstudio.com/resources/cheatsheets/
[3] https://r4ds.had.co.nz/data-visualisation.html
[4] https://www.sohu.com/a/320024110_718302

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

友情鏈接更多精彩內(nèi)容