1. 程式人生 > >Python數據可視化的四種簡易方法

Python數據可視化的四種簡易方法

陰影 cat 一點 代碼 概率分布 man sda stat word

摘要: 本文講述了熱圖、二維密度圖、蜘蛛圖、樹形圖這四種Python數據可視化方法。

數據可視化是任何數據科學或機器學習項目的一個重要組成部分。人們常常會從探索數據分析(EDA)開始,來深入了解數據,並且創建可視化確實有助於讓問題更清晰和更容易理解,尤其是對於那些較大的高維度數據集。在項目結束的時候,能夠以清晰的、簡潔的和令人信服的方式呈現最終結果,這是非常重要的,讓你的用戶能夠理解和明白。

你可能已經看過了我之前的文章《5種快速和簡單的Python數據可視化方法(含代碼)》(5 Quick and Easy Data Visualizations in Python with Code),其中介紹了5種基本可視化方法:散點圖、線圖、柱狀圖、條形圖和箱形圖。這五個是簡單而強大的可視化方法,你絕對可以通過這些方法從數據集中得到巨大的收獲。在本文中,將介紹另外4個數據可視化方法,但稍微復雜一些,你可以在看完上一篇文章介紹的基本方法之後再用。

熱圖(Heat Map)

熱圖是數據的矩陣表示方式,其中每個矩陣的值用一種顏色來表示。不同的顏色代表不同的級別,矩陣指數將兩個對比的列或特征連接在一起。熱圖可以很好地顯示出多個特征變量之間的關系,因為可以直接把一個級別看作一種顏色。還可以通過觀察熱圖中的一些點來查看每個關系是如何與數據集中的其它關系進行比較的。這些顏色的確提供了簡單的表示方式,因為這是非常直觀的。

技術分享圖片

現在來看下代碼:與matplotlib庫相比,seaborn庫可用於更高級的圖表,通常也需要更多的組件,如更多的顏色、圖形或者變量。Matplotlib庫用於顯示圖表,numpy用於生成數據,而pandas用於控制。繪圖只是調用一個簡單的seaborn函數,如果你發現了一些在視覺上很特別的東西,通過這個函數,還可以設置顏色映射。

# Importing libs
importseaborn as sns
import pandas aspd
importnumpyasnp
importmatplotlib.pyplotasplt

# Create a random dataset
data=pd.DataFrame(np.random.random((10,6)), columns=["Iron Man","CaptainAmerica","BlackWidow","Thor","Hulk", "Hawkeye"])

print(data)

# Plot the heatmap
heatmap_plot=sns.heatmap(data, center=0, cmap=‘gist_ncar‘)

plt.show()

二維密度圖(2D Density Plot)

二維密度圖是一維版本的簡單擴展,能夠看到關於2個變量的概率分布。讓我們看看下面的二維密度圖,右邊的刻度用顏色表示每一點的概率。最高的概率,看下數據集,似乎大約是0.5的大小和1.4-ish的速度。正如你所看到的,二維密度圖對於快速確定數據對於兩個變量最集中的區域非常地顯著,而不是像一維密度圖那樣只集中一個變量。當你有兩個對輸出結果非常重要的變量,並且希望了解它們如何一起對輸出結果分布起作用的時候,二維密度圖尤其適合。

技術分享圖片

Seaborn的代碼超級簡單,我們將通過創建一個偏態分布介紹它。如果你發現某些顏色和陰影在視覺上更特別,那麽大多數的可選參數都是為了看起來更清晰。

蜘蛛圖(Spider Plot)

蜘蛛圖是顯示一對多關系最好的方法之一。也就是說,你可以繪制並查看區別於單個變量或類別的多個變量的值。在蜘蛛圖中,一個變量相對於另一個變量的特性是顯而易見的,因為面積和長度在一些方向上變化了。如果你希望了解幾個類別關於這些變量是如何疊加起來的,可以並排繪制一下。在下圖中,很容易比較三個電影角色的不同屬性,並了解他們的優勢所在!

技術分享圖片

這次我們將能夠直接使用matplotlib來創建可視化,而不是用seaborn。需要計算每個屬性所在的角度,因為我們希望它們沿圓周被平均地分隔開。我們將在每個計算的角度放置標簽,然後把值繪制成一個點,該點到中心的距離取決於它的值或是級別。最後,為了清晰起見,我們將使用半透明的顏色填充由連接各屬性點的線所包含的區域。

# Import libs
import pandas aspd
importseabornassns
importnumpyasnp
importmatplotlib.pyplotasplt

# Get the data
df=pd.read_csv("avengers_data.csv")
print(df)

"""
   #             Name  Attack  Defense  Speed  Range  Health
0  1         Iron Man      83       80     75     70      70
1  2  Captain America      60       62     63     80      80
2  3             Thor      80       82     83    100     100
3  3             Hulk      80      100     67     44      92
4  4      Black Widow      52       43     60     50      65
5  5          Hawkeye      58       64     58     80      65

"""

# Get the data for Iron Man
labels=np.array(["Attack","Defense","Speed","Range","Health"])
stats=df.loc[0,labels].values

# Make some calculations for the plot
angles=np.linspace(0, 2*np.pi, len(labels), endpoint=False)
stats=np.concatenate((stats,[stats[0]]))
angles=np.concatenate((angles,[angles[0]]))

# Plot stuff
fig=plt.figure()
ax=fig.add_subplot(111, polar=True)
ax.plot(angles, stats, ‘o-‘, linewidth=2)
ax.fill(angles, stats, alpha=0.25)
ax.set_thetagrids(angles *180/np.pi, labels)
ax.set_title([df.loc[0,"Name"]])
ax.grid(True)

plt.show()

樹形圖(Tree Diagram)

我們從小學就開始使用樹形圖了,樹形圖既自然又直觀,還易於解釋。直接連接的節點關系密切,而與有多個連接的節點差別很大。在下圖中,我已經根據統計繪制了一小部分來自Kaggle的Pokemon with stats數據集:

HP、攻擊、防禦、特殊攻擊、特殊防禦、速度

因此,與stats wise最匹配的Pokemon將緊密連接在一起。例如,我們看到,在頂部,Arbok和Fearow是直接連接的,而且,如果我們查看數據,Arbok總共有438個,而Fearow有442個,非常接近。但是一旦我們移動到Raticate,我們得到的總數是413,這與Arbok和Fearow的差別很大,這就是它們被分開的原因。當我們移動樹的時候,基於相似性,Pokemon被分的組越來越多。在綠色組中的Pokemon相互之間比紅色組中的更相似,即使沒有直接的綠色連接。

技術分享圖片

對於樹形圖,我們實際上要使用Scipy的。在查看了數據集之後,我們將去掉字符串類型的列。我們這麽做只是為了要得到正確的可視化結果,但在實踐中,最好是把這些字符串轉換成分類變量,為了得到更好的結果和進行比較,我們還設置了數據幀索引,以便能夠適當地用它作為引用每個節點的列。最後,在Scipy中計算和繪制樹形圖是非常簡單的事了。

# Import libs
import pandas aspd
frommatplotlibimportpyplotasplt
fromscipy.clusterimport hierarchy
importnumpyasnp

# Read in the dataset
# Drop any fields that are strings
# Only get the first 40 because this dataset is big
df=pd.read_csv(‘Pokemon.csv‘)
df=df.set_index(‘Name‘)
del df.index.name
df=df.drop(["Type 1", "Type 2", "Legendary"], axis=1)
df=df.head(n=40)

# Calculate the distance between each sample
Z =hierarchy.linkage(df, ‘ward‘)

# Orientation our tree
hierarchy.dendrogram(Z, orientation="left", labels=df.index)

plt.show()

推薦閱讀

為了了解更多的關於數據可視化方面的知識,我建議大家學習這本書 — Data Visualisation Book,它提供了關於何時、何地、以及為什麽使用各個類型的可視化方法的全面而直觀的講解。

原文鏈接

本文為雲棲社區原創內容,未經允許不得轉載。

Python數據可視化的四種簡易方法