以列值為散佈圖著色
在Python 中,Matplotlib 函式庫提供了多種自訂散佈圖美觀的方法。一項常見任務是根據特定列中的值指派顏色。
Seaborn 整合
一個解決方案是利用基於 Matplotlib 建構的 Seaborn 函式庫。 Seaborn 提供 sns.relplot 和 sns.FacetGrid 等進階函數,讓您可以輕鬆地將散佈圖對應到特定列。透過指定hue參數,您可以根據包含類別標籤的第三列對點進行著色。
import seaborn as sns
sns.relplot(data=df, x='Weight (kg)', y='Height (cm)', hue='Gender')
直接使用Matplotlib
或者,您可以直接使用Matplotlib 的plt.scatter 函數建立散佈圖並手動指定顏色。這需要建立一個自訂顏色字典,將類別標籤對應到顏色。
def dfScatter(df, xcol='Height', ycol='Weight', catcol='Gender'):
fig, ax = plt.subplots()
categories = np.unique(df[catcol])
colors = np.linspace(0, 1, len(categories))
colordict = dict(zip(categories, colors))
df['Color'] = df[catcol].apply(lambda x: colordict[x])
ax.scatter(df[xcol], df[ycol], c=df.Color)
return fig
透過呼叫函數,可以產生指定類別列著色的散佈圖:
df = pd.DataFrame({'Height': np.random.normal(size=10),
'Weight': np.random.normal(size=10),
'Gender': ["Male", "Male", "Unknown", "Male", "Male",
"Female", "Did not respond", "Unknown", "Female", "Female"]})
fig = dfScatter(df)
免責聲明: 提供的所有資源部分來自互聯網,如果有侵犯您的版權或其他權益,請說明詳細緣由並提供版權或權益證明然後發到郵箱:[email protected] 我們會在第一時間內為您處理。
Copyright© 2022 湘ICP备2022001581号-3