Bagaimana untuk Mewarnai Plot Taburan mengikut Nilai Lajur dalam Python?

Mewarnai Plot Taburan mengikut Nilai Lajur

Dalam Python, perpustakaan Matplotlib menyediakan beberapa cara untuk menyesuaikan estetika plot taburan. Satu tugas biasa ialah menetapkan warna berdasarkan nilai dalam lajur tertentu.

Integrasi Seaborn

Satu penyelesaian ialah memanfaatkan perpustakaan Seaborn, yang dibina di atas Matplotlib. Seaborn menawarkan fungsi peringkat tinggi seperti sns.relplot dan sns.FacetGrid yang membolehkan anda memetakan plot taburan dengan mudah pada lajur tertentu. Dengan menentukan parameter rona, anda boleh mewarnakan titik mengikut lajur ketiga yang mengandungi label kategori.

<code class="python">import seaborn as sns

sns.relplot(data=df, x='Weight (kg)', y='Height (cm)', hue='Gender')</code>

Secara Terus Menggunakan Matplotlib

Sebagai alternatif, anda boleh terus menggunakan Matplotlib's fungsi plt.scatter untuk membuat plot serakan dan menentukan warna secara manual. Ini memerlukan mencipta kamus warna tersuai yang memetakan label kategori kepada warna.

<code class="python">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</code>

Dengan memanggil fungsi ini, anda boleh menjana plot taburan yang diwarnakan mengikut lajur kategori yang ditentukan:

<code class="python">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)</code>

Atas ialah kandungan terperinci Bagaimana untuk Mewarnai Plot Taburan mengikut Nilai Lajur dalam Python?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!