資料視覺化乾貨：使用pandas和seaborn製作炫酷圖表（附程式碼）

作者：Wes McKinney

本文摘編自《利用Python進行資料分析》（原書第2版），如需轉載請聯絡我們

01 折線圖

Series和DataFrame都有一個plot屬性，用於繪製基本的圖型。預設情況下，plot()繪製的是折線圖(見圖9-13)：

▲圖9-13 簡單序列圖形

Series物件的索引傳入matplotlib作為繪圖的x軸，你可以透過傳入use_index=False來禁用這個功能。x軸的刻度和範圍可以透過xticks和xlim選項進行調整，相應地y軸使用yticks和ylim進行調整。表9-3是plot的全部選項串列。本節我會介紹這些選項中的一些，其餘你可以自行探索。

大部分pandas的繪圖方法，接收可選的ax引數，該引數可以是一個matplotlib子圖物件。這使你可以更為靈活的在網格佈局中放置子圖。

DataFrame的plot方法在同一個子圖中將每一列繪製為不同的折線，並自動生成圖例(見圖9-14)：

▲圖9-14 簡單DataFrame繪圖

plot屬性包含了不同繪圖型別的方法族。例如，df.plot( )等價於df.plot.line( )。我們之後將會探索這些方法中的一部分。

要繪製的其他關鍵字引數會傳遞到相應的matplotlib繪圖函式，因此你可以透過瞭解更多的matplotlib的 API資訊來進一步定製這些圖表。

▲表9-3 Series.plot方法引數

DataFrame擁有多個選項，允許靈活地處理列；例如，是否將各列繪製到同一個子圖中，或為各列生成獨立的子圖。參考表9-4瞭解更多選項。

▲表9-4

02 柱狀圖

plot.bar()和plot.barh()可以分別繪製垂直和水平的柱狀圖。在繪製柱狀圖時，Series或DataFrame的索引將會被用作x軸刻度(bar)或y軸刻度(barh)(參考圖9-15)：

▲圖9-15 水平柱狀圖和垂直柱狀圖

選項color=’k’和alpha=0.7將柱子的顏色設定為黑色，並將影象的填充色設定為部分透明。

在DataFrame中，柱狀圖將每一行中的值分組到併排的柱子中的一組。參考圖9-16：

▲圖9-16 DataFrame柱狀圖

請註意DataFrame的列名稱”Genus”被用作了圖例標題。我們可以透過傳遞stacked=True來生成堆積柱狀圖，會使得每一行的值堆積在一起(參考圖9-17)：

▲圖9-17 DataFrame堆積柱狀圖

使用value_counts: s.value_counts().plot.bar()可以有效的對Series值頻率進行視覺化。

回到本書之前使用的資料集，假設我們想要繪製一個堆積柱狀圖，用於展示每個派對在每天的資料點佔比。使用read_csv載入資料，並根據星期幾數值和派對規模進形成交叉表：

之後，進行標準化以確保每一行的值和為1，然後進行繪圖(見圖9-18)：

▲圖9-18 每天派對數量的百分比

你可以看到本資料集中的派對數量在週末會增加。

對於在繪圖前需要聚合或彙總的資料，使用seaborn包會使工作更為簡單。現在讓我們看下使用seaborn進行按星期幾數值計算小費百分比(見圖9-19中的結果圖)：

▲圖9-19 用錯誤欄按天顯示小費百分比

seaborn中的繪圖函式使用一個data引數，這個引數可以是pandas的DataFrame。其他的引數則與列名有關。因為day列中有多個觀測值，柱子的值是tip_pct的平均值。柱子上畫出的黑線代表的是95%的置信區間(置信區間可以透過可選引數進行設定)。

seaborn.barplot擁有一個hue選項，允許我們透過一個額外的分類值將資料分離：

▲圖9-20 根據星期幾數值和時間計算的小費百分比

請註意seaborn自動改變了圖表的美觀性：預設的調色盤、圖背景和網格線條顏色。你可以使用seaborn.set在不同的繪圖外觀中進行切換：

03 直方圖和密度圖

直方圖是一種條形圖，用於給出值頻率的離散顯示。資料點被分成離散的，均勻間隔的箱，並且繪製每個箱中資料點的數量。使用之前的小費資料，我們可以使用Series的plot.hist方法製作小費佔總費用百分比的直方圖(見圖9-21)：

▲圖9-21 小費百分比的直方圖

密度圖是一種與直方圖相關的圖表型別，它透過計算可能產生觀測資料的連續機率分佈估計而產生。通常的做法是將這種分佈近似為“核心”的混合，也就是像正態分佈那樣簡單的分佈。因此，密度圖也被成為核心密度估計圖(KDE)。plot.kde使用傳統法定混合法估計繪製密度圖(見圖9-22)：

▲圖9-22 小費百分比密度圖

distplot方法可以繪製直方圖和連續密度估計，透過distplot方法seaborn使直方圖和密度圖的繪製更為簡單。作為例子，考慮由兩個不同的標準正態分佈組成的雙峰分佈(見圖9-23)：

▲圖9-23 正態混合的標準化直方圖與密度估計

04 散點圖或點圖

點圖或散點圖可以用於檢驗兩個一維資料序列之間的關係。例如，這裡我們從statsmodels專案中載入了macrodata資料集，並選擇了一些變數，之後計算對數差：

然後我們可以使用seaborn的reglot方法，該方法可以繪製散點圖，並擬合出一個條線性回歸線(見圖9-24)：

▲圖9-24 seaborn回歸/散點圖

在探索性資料分析中，能夠檢視一組變數中的所有散點圖是有幫助的; 這被稱為成對圖或散點圖矩陣。從頭開始繪製這樣一個圖是有點工作量的，所以seaborn有一個方便的成對圖函式，它支援在對角線上放置每個變數的直方圖或密度估計值(結果圖見圖9-25)：

▲圖9-25 statsmodels macro資料的成對圖矩陣

你可能會註意到plot_ksw引數，這個引數使我們能夠將配置選項傳遞給非對角元素上的各個繪圖呼叫。參考seaborn.pairplot的檔案字串可以看到更多細節的設定選項。

05 分面網格和分類資料

如果資料集有額外的分組維度怎麼辦？使用分面網格是利用多種分組變數對資料進行視覺化的方式。seaborn擁有一個有效的內建函式factorplot，可以簡化多種分面繪圖(見圖9-26)：

▲圖9-26 按星期幾數值/時間/是否吸煙劃分的小費百分比

除了根據’time’在一個面內將不同的柱分組為不同的顏色，我們還可以透過每個時間值新增一行來擴充套件分面網格(見圖9-27)：

▲圖9-27 根據時間/是否吸煙分面後按星期幾數值劃分的小費百分比

factorplot 支援其他可能有用的圖型別，具體取決於你要顯示的內容。 例如，箱形圖（顯示中位值，四分位數和異常值）可以是有效的視覺化型別(圖9-28)：

▲圖9-28 根據星期幾數值繪製的小費百分比箱型圖

你可以使用更通用的seaborn.FacetGrid類建立自己的分面網格圖。 具體請檢視更多的seaborn檔案。

06 其他Python視覺化工具

和開原始碼一樣，在Python語言下建立圖形的選擇有很多(太多而無法一一列舉)。自從2010年以來，很多開發工作都集中在建立web互動式圖形上。藉助像Bokeh和Plotly這樣的工具，在web瀏覽器中建立動態的、互動式影象的工作現在已經可以實現。

如果是建立用於印刷或網頁的靜態圖形，我建議根據你的需要使用預設的matplotlib以及像pandas和seaborn這樣的附加庫。 對於其他資料視覺化要求，學習其他可用工具之一可能是有用的。我鼓勵你探索Python視覺化生態系統，因為它將持續增添新內容併在未來進行更多創新。

關於作者：韋斯·麥金尼（Wes McKinney）是流行的Python開源資料分析庫pandas的創始人。他是一名活躍的演講者，也是Python資料社群和Apache軟體基金會的Python/C++開源開發者。目前他在紐約從事軟體架構師工作。

本文摘編自《利用Python進行資料分析》（原書第2版），經出版方授權釋出。

延伸閱讀《利用Python進行資料分析》

點選上圖瞭解及購買

轉載請聯絡微信：togo-maruko

推薦語：Python資料分析經典暢銷書全新升級，第1版中文版累計銷售100000冊。針對Python 3.6進行全面修訂和更新，涵蓋新版的pandas、NumPy、IPython和Jupyter。