Graphs & Plots in Data Science

Graphs and plots are essential tools in Data Science for visually analyzing and communicating data. They help in understanding patterns, trends, relationships, and distributions.

1. Types of Graphs & Plots in Data Science

1.1. Line Plot

Used to display trends over time or continuous data.
Best suited for time series analysis or showing changes over intervals.

Example: Plotting a Simple Line Graph

import matplotlib.pyplot as plt
import numpy as np

# Generate sample data
x = np.linspace(0, 10, 100)
y = np.sin(x)

# Create a line plot
plt.plot(x, y, label="Sine Wave", color="blue")
plt.title("Line Plot Example")
plt.xlabel("X-axis")
plt.ylabel("Y-axis")
plt.legend()
plt.show()

Try It Now

1.2. Bar Chart

Used for comparing categorical data.
Shows differences between groups or categories.

Example: Visualizing Sales Data with a Bar Chart

import seaborn as sns
import pandas as pd

# Sample data
data = pd.DataFrame({'Product': ['A', 'B', 'C', 'D'], 'Sales': [120, 340, 230, 410]})

# Create a bar chart
sns.barplot(x='Product', y='Sales', data=data)
plt.title("Product Sales Comparison")
plt.show()

Try It Now

1.3. Histogram

Used to show the distribution of numerical data.
Useful for understanding the frequency of data within certain ranges.

Example: Plotting a Histogram of Student Scores

import numpy as np
import matplotlib.pyplot as plt

# Generate random data
scores = np.random.normal(70, 10, 1000)

# Create a histogram
plt.hist(scores, bins=20, edgecolor="black")
plt.title("Distribution of Student Scores")
plt.xlabel("Scores")
plt.ylabel("Frequency")
plt.show()

Try It Now

1.4. Scatter Plot

Used to visualize relationships between two numerical variables.
Helps in identifying correlations and trends.

Example: Scatter Plot of Age vs. Salary

import numpy as np
import pandas as pd
import seaborn as sns

# Generate random data
np.random.seed(10)
data = pd.DataFrame({
    'Age': np.random.randint(20, 60, 100),
    'Salary': np.random.randint(30000, 100000, 100)
})

# Create scatter plot
sns.scatterplot(x='Age', y='Salary', data=data)
plt.title("Age vs. Salary Relationship")
plt.show()

Try It Now

1.5. Box Plot

Used to visualize the distribution, median, and outliers in numerical data.
Helps in detecting anomalies.

Example: Box Plot of Monthly Salaries

import seaborn as sns
import numpy as np
import pandas as pd

# Generate sample data
np.random.seed(42)
salary_data = pd.DataFrame({
    'Department': ['HR', 'IT', 'Finance', 'Marketing', 'HR', 'IT', 'Finance', 'Marketing'],
    'Salary': np.random.randint(40000, 120000, 8)
})

# Create a box plot
sns.boxplot(x='Department', y='Salary', data=salary_data)
plt.title("Salary Distribution by Department")
plt.show()

Try It Now

1.6. Pie Chart

Used to show proportions of different categories.
Best for visualizing percentage distributions.

Example: Pie Chart of Market Share Distribution

import matplotlib.pyplot as plt

# Sample data
labels = ['Brand A', 'Brand B', 'Brand C', 'Brand D']
sizes = [30, 25, 20, 25]
colors = ['blue', 'green', 'red', 'purple']

# Create pie chart
plt.pie(sizes, labels=labels, autopct='%1.1f%%', colors=colors, startangle=90)
plt.title("Market Share Distribution")
plt.show()

Try It Now

1.7. Heatmap

Used to show relationships between multiple numerical variables using color intensity.
Commonly used for correlation matrices.

Example: Visualizing Correlation Between Features

import seaborn as sns
import pandas as pd
import numpy as np

# Generate sample data
np.random.seed(42)
data = pd.DataFrame({
    'A': np.random.rand(10),
    'B': np.random.rand(10),
    'C': np.random.rand(10),
    'D': np.random.rand(10)
})

# Compute correlation matrix
corr_matrix = data.corr()

# Create a heatmap
sns.heatmap(corr_matrix, annot=True, cmap='coolwarm', linewidths=0.5)
plt.title("Feature Correlation Heatmap")
plt.show()

Try It Now

2. Choosing the Right Graph for Your Data

Graph Type	Best For	Example
Line Plot	Trends over time	Stock price movement, temperature changes
Bar Chart	Comparing categories	Sales comparison, population by country
Histogram	Data distribution	Exam scores, age distribution
Scatter Plot	Relationship between variables	Age vs. salary, height vs. weight
Box Plot	Data distribution and outliers	Salary distribution, test scores
Pie Chart	Proportional comparison	Market share, percentage distribution
Heatmap	Correlation between variables	Feature relationships in datasets

Summary

Line plots are great for time series data.
Bar charts help compare categorical data.
Histograms display distributions of numerical values.
Scatter plots reveal relationships between two numerical variables.
Box plots highlight outliers and distribution summaries.
Pie charts illustrate percentage breakdowns.
Heatmaps show correlations between multiple numerical variables.

Understanding these visualization techniques is essential for effective data analysis and storytelling.