पांडा और मैटप्लोटलिब का उपयोग करके कॉलम मानों द्वारा स्कैटर प्लॉट्स को रंगना

मैटप्लोटलिब स्थिर, एनिमेटेड और इंटरैक्टिव विज़ुअलाइज़ेशन बनाने के लिए एक लोकप्रिय पायथन लाइब्रेरी है अजगर. यह आलेख पांडा डेटाफ़्रेम के एक विशिष्ट कॉलम में मानों के आधार पर स्कैटर प्लॉट को रंगने के लिए मैटप्लोटलिब का उपयोग करने की पड़ताल करता है।

आयात और डेटा

शुरू करने के लिए, हम आवश्यक लाइब्रेरी आयात करते हैं , जिसमें मैटप्लोटलिब (पीएलटी के रूप में) और पांडा (पीडी के रूप में) शामिल हैं। हम तीन कॉलमों के साथ एक नमूना डेटाफ़्रेम ("डीएफ") भी बनाते हैं: "ऊंचाई," "वजन," और "लिंग।"

import matplotlib.pyplot as plt
import pandas as pd
import numpy as np

np.random.seed(0)
N = 37
_genders = ["Female", "Male", "Non-binary", "No Response"]
df = pd.DataFrame({
    "Height (cm)": np.random.uniform(low=130, high=200, size=N),
    "Weight (kg)": np.random.uniform(low=30, high=100, size=N),
    "Gender": np.random.choice(_genders, size=N),
})

अगस्त 2021 में अपडेट हो रहा है

सीबॉर्न ने संस्करण 0.11.0 में सीबॉर्न.relplot जैसे नए फिगर-स्तरीय फ़ंक्शन पेश किए हैं। सीधे FacetGrid का उपयोग करने पर इन कार्यों की अनुशंसा की जाती है। रंग_क्रम=_लिंग, पहलू=1.61) plt.show()

sns.relplot(data=df, x="Weight (kg)", y="Height (cm)", hue="Gender", hue_order=_genders, aspect=1.61)
plt.show()

यदि आप सीधे मैटप्लोटलिब का उपयोग करना चाहते हैं, तो आपको मैटप्लोटलिब के स्कैटर फ़ंक्शन को पांडास डेटाफ़्रेम की श्रेणियों पर मैप करना होगा . ऐसा करने के लिए:

कॉलम और रंगों से अद्वितीय श्रेणियों के साथ एक शब्दकोश बनाएं।

डेटाफ़्रेम में एक नया "रंग" कॉलम जोड़ें, प्रत्येक श्रेणी को एक संबंधित रंग निर्दिष्ट करें।
• रंग कॉलम को "सी" के रूप में निर्दिष्ट करते हुए, डेटा प्लॉट करने के लिए स्कैटर फ़ंक्शन का उपयोग करें तर्क।
• def dfScatter(df, xcol='ऊंचाई', ycol='वजन', catcol='लिंग'): अंजीर, कुल्हाड़ी = plt.subplots() श्रेणियाँ = np.unique(df[catcol]) रंग = np.linspace(0, 1, len(श्रेणियाँ)) colordict = dict(ज़िप(श्रेणियाँ, रंग)) डीएफ["रंग"] = डीएफ[कैटकॉल].लागू करें(लैम्ब्डा एक्स: कलरडिक्ट[एक्स]) ax.scatter(df[xcol], df[ycol], c=df.Color) वापसी अंजीर अंजीर = डीएफस्कैटर(डीएफ) fig.savefig('fig1.png')

इन चरणों का पालन करके, आप पांडा और मैटप्लोटलिब का उपयोग करके कॉलम मानों के आधार पर स्कैटर प्लॉट को आसानी से रंग सकते हैं।