選自towardsdatascience
作者：George Seif
機器之心編譯
參與：Geek AI、劉曉坤
熱力圖、二維密度圖、蜘蛛網圖和樹狀圖，這些可視化方法你都用過嗎？
數據可視化是數據科學或機器學習項目中十分重要的一環。通常，你需要在項目初期進行探索性的數據分析（EDA），從而對數據有一定的了解，而且創建可視化確實可以使分析的任務更清晰、更容易理解，特別是對於大規模的高維數據集。在項目接近尾聲時，以一種清晰、簡潔而引人注目的方式展示最終結果也是非常重要的，讓你的受眾（通常是非技術人員的客戶）能夠理解。
讀者可能閱讀過我之前的文章「5 Quick and Easy Data Visualizations in Python with Code」，我通過那篇文章向大家介紹了 5 種基礎的數據可視化方法：散點圖、線圖、直方圖、條形圖和箱形圖。這些都是簡單而強大的可視化方法，通過它們你可以對數據集有深刻的認識。在本文中，我們將看到另外 4 個數據可視化方法！本文對這些方法的介紹會更詳細一些，可以在您閱讀了上一篇文章中的基本方法之後接著使用，從而從數據中提取出更深入的信息。
熱力圖
熱力圖（Heat Map）是數據的一種矩陣表示方法，其中每個矩陣元素的值通過一種顏色表示。不同的顏色代表不同的值，通過矩陣的索引將需要被對比的兩項或兩個特徵關聯在一起。熱力圖非常適合於展示多個特徵變量之間的關係，因為你可以直接通過顏色知道該位置上的矩陣元素的大小。通過查看熱力圖中的其他點，你還可以看到每種關係與數據集中的其它關係之間的比較。顏色是如此直觀，因此它為我們提供了一種非常簡單的數據解釋方式。
現在讓我們來看看實現代碼。與「matplotlib」相比，「seaborn」可以被用於繪製更加高級的圖形，它通常需要更多的組件，例如多種顏色、圖形或變量。「matplotlib」可以被用於顯示圖形，「NumPy」可被用於生成數據，「pandas」可以被用於處理數據！繪圖只是「seaborn」的一個簡單的功能。
import seaborn as sns
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
data = pd.DataFrame(np.random.random((10,6)), columns=["Iron Man","Captain America","Black Widow","Thor","Hulk", "Hawkeye"])
print(data)
heatmap_plot = sns.heatmap(data, center=0, cmap='gist_ncar')
plt.show()
二維密度圖
二維密度圖（2D Density Plot）是一維版本密度圖的直觀擴展，相對於一維版本，其優點是能夠看到關於兩個變量的概率分布。例如，在下面的二維密度圖中，右邊的刻度圖用顏色表示每個點的概率。我們的數據出現概率最大的地方（也就是數據點最集中的地方），似乎在 size=0.5，speed=1.4 左右。正如你現在所知道的，二維密度圖對於迅速找出我們的數據在兩個變量的情況下最集中的區域非常有用，而不是像一維密度圖那樣只有一個變量。當你有兩個對輸出非常重要的變量，並且希望了解它們如何共同作用於輸出的分布時，用二維密度圖觀察數據是十分有效的。
事實再次證明，使用「seaborn」編寫代碼是十分便捷的！這一次，我們將創建一個偏態分布，讓數據可視化結果更有趣。你可以對大多數可選參數進行調整，讓可視化看結果看起來更清楚。
import seaborn as sns
import matplotlib.pyplot as plt
from scipy.stats import skewnorm
speed = skewnorm.rvs(4, size=50)
size = skewnorm.rvs(4, size=50)
ax = sns.kdeplot(speed, size, cmap="Reds", shade=False, bw=.15, cbar=True)
ax.set(xlabel='speed', ylabel='size')
plt.show()
蜘蛛網圖
蜘蛛網圖（Spider Plot）是顯示一對多關係的最佳方法之一。換而言之，你可以繪製並查看多個與某個變量或類別相關的變量的值。在蜘蛛網圖中，一個變量相對於另一個變量的顯著性是清晰而明顯的，因為在特定的方向上，覆蓋的面積和距離中心的長度變得更大。如果你想看看利用這些變量描述的幾個不同類別的對象有何不同，可以將它們並排繪製。在下面的圖表中，我們很容易比較復仇者聯盟的不同屬性，並看到他們各自的優勢所在！（請注意，這些數據是隨機設置的，我對復仇者聯盟的成員們沒有偏見。）
在這裡，我們可以直接使用「matplotlib」而非「seaborn」來創建可視化結果。我們需要讓每個屬性沿圓周等距分布。我們將在每個角上設置標籤，然後將值繪製為一個點，它到中心的距離取決於它的值/大小。最後，為了顯示更清晰，我們將使用半透明的顏色來填充將屬性點連接起來得到的線條所包圍的區域。
import pandas as pd
import seaborn as sns
import numpy as np
import matplotlib.pyplot as plt
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
"""
labels=np.array(["Attack","Defense","Speed","Range","Health"])
stats=df.loc[0,labels].values
angles=np.linspace(0, 2*np.pi, len(labels), endpoint=False)
stats=np.concatenate((stats,[stats[0]]))
angles=np.concatenate((angles,[angles[0]]))
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 的統計數據（生命值、攻擊力、防禦力、特殊攻擊、特殊防禦、速度）繪製了一小部分口袋妖怪遊戲的數據集的樹狀圖。
因此，統計意義上最匹配的口袋妖怪將被緊密地連接在一起。例如，在圖的頂部，阿柏怪 和尖嘴鳥是直接連接的，如果我們查看數據，阿柏怪的總分為 438，尖嘴鳥則為 442，二者非常接近！但是如果我們看看拉達，我們可以看到其總得分為 413，這和阿柏怪、尖嘴鳥就具有較大差別了，所以它們在樹狀圖中是被分開的！當我們沿著樹往上移動時，綠色組的口袋妖怪彼此之間比它們和紅色組中的任何口袋妖怪都更相似，即使這裡並沒有直接的綠色的連接。
對於樹狀圖，我們實際上需要使用「Scipy」來繪製！讀取數據集中的數據之後，我們將刪除字符串列。這麼做只是為了使可視化結果更加直觀、便於理解，但在實踐中，將這些字符串轉換為分類變量會得到更好的結果和對比效果。我們還設置了數據幀的索引，以便能夠恰當地將其用作引用每個節點的列。最後需要告訴大家的是，在「Scipy」中計算和繪製樹狀圖只需要一行簡單的代碼。
import pandas as pd
from matplotlib import pyplot as plt
from scipy.cluster import hierarchy
import numpy as np
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)
Z = hierarchy.linkage(df, 'ward')
hierarchy.dendrogram(Z, orientation="left", labels=df.index)
plt.show()
原文連結：https://towardsdatascience.com/4-more-quick-and-easy-data-visualizations-in-python-with-code-da9030ab3429
本文為機器之心編譯，轉載請聯繫本公眾號獲得授權。
✄---
加入機器之心（全職記者 / 實習生）：hr@jiqizhixin.com
投稿或尋求報導：content@jiqizhixin.com
廣告 & 商務合作：bd@jiqizhixin.com