在Python中調用ggplot的三種方法

2019-11-25 17:47:58

字體：大中小

來源：轉載

供稿：網友

本文提供了三種不同的方式在Python（IPython Notebook）中調用ggplot。

在大數據時代，數據可視化是一個非常熱門的話題。各個BI的廠商無不在數據可視化領域里投入大量的精力。Tableau憑借其強大的數據可視化的功能成為硅谷炙手可熱的上市公司。Tableau的數據可視化的產品，其理論基礎其實是《The Grammar of Graphic》，該書提出了對信息可視化的圖表的語法抽象體系，數據的探索和分析可以由圖像的語法來驅動，而非有固定的圖表類型來驅動，使得數據的探索過程變得友好而有趣。

然而對于The Grammar of Graphic的理論的實踐，并非Tableau獨占，ggplot作為R語言上得一個圖形庫，其理論基礎也是這本書。（注，筆者曾就職的某BI巨頭，主要職責也是數據可視化，我們曾經和加拿大團隊研發過類似的產品，基于HTML5和D3，可惜由于種種原因未能推向市場）

現在越來越多的人開始使用python來做數據分析，IPython Notebook尤其令人喜愛，它的實時交互把腳本語言的優勢發揮到極致。那么怎樣才能在IPython Notebook中使用ggplot呢？我這里跟大家分享三種不同的方式供大家選擇。
RPy2

第一種方式是使用rpy2, rpy2是對rpy的改寫和重新設計，旨在提供Python用戶在python中使用R的API。

rpy2提供了對R語言的對象和方法的基本封裝，當然也包括可視化的圖庫這一塊。

下面就是一段運行ggplot的R程序使用rpy2在python中運行的例子：

from rpy2 import robjectsfrom rpy2.robjects import Formula, Environmentfrom rpy2.robjects.vectors import IntVector, FloatVectorfrom rpy2.robjects.lib import gridfrom rpy2.robjects.packages import importr, dataimport rpy2.robjects.lib.ggplot2 as ggplot2 # The R 'print' functionrprint = robjects.globalenv.get("print")stats = importr('stats')grdevices = importr('grDevices')base = importr('base')datasets = importr('datasets') mtcars = data(datasets).fetch('mtcars')['mtcars'] pp = ggplot2.ggplot(mtcars) + /   ggplot2.aes_string(x='wt', y='mpg', col='factor(cyl)') + /   ggplot2.geom_point() + /   ggplot2.geom_smooth(ggplot2.aes_string(group = 'cyl'),             method = 'lm')pp.plot()

以上程序在IPython Notebook中運行會有缺陷，會彈出一個新的窗口顯示圖，而且該python進程會阻塞在那里。我們希望圖表能內嵌在IPython Notebook的頁面中，為了解決該問題，我們引入如下代碼：

%matplotlib inline import uuidfrom rpy2.robjects.packages import importr from IPython.core.display import Image grdevices = importr('grDevices')def ggplot_notebook(gg, width = 800, height = 600):  fn = '{uuid}.png'.format(uuid = uuid.uuid4())  grdevices.png(fn, width = width, height = height)  gg.plot()  grdevices.dev_off()  return Image(filename=fn)

運行上述代碼后，我們把ggplot的調用pp.plot()改為調用ggplot_notebook(pp, height=300)就能成功嵌入顯示ggplot的結果。

201548145241359.png (800×300)

RMagic

另一種方式是使用rmagic，rmagicy實際上依賴于rpy2。它的使用方式更像是直接在使用R

%load_ext rmagiclibrary(ggplot2)dat <- data.frame(x = rnorm(10), y = rnorm(10),          lab = sample(c('A', 'B'), 10, replace = TRUE))x <- ggplot(dat, aes(x = x, y = y, color = lab)) + geom_point()print(x)

運行結果如下

201548145327917.png (480×480)

ggplot for python

ggplot是一個python的庫，基本上是對R語言ggplot的功能移植到Python上。

運行安裝腳本

pip install ggplot

安裝成功后，可以試一下這個例子

%matplotlib inlineimport pandas as pdfrom ggplot import *meat_lng = pd.melt(meat[['date', 'beef', 'pork', 'broilers']], id_vars='date')ggplot(aes(x='date', y='value', colour='variable'), data=meat_lng) + /  geom_point() + /  stat_smooth(color='red')

結果如下：

201548145402721.png (649×499)