6.5. Visual Exploration with Matplotlib and Seaborn

import matplotlib.pyplot as plt
df['Age'].hist(bins=10)
plt.title("Age Distribution")
plt.show()

A histogram is a graphical representation that organizes a group of data points into user-specified ranges. In this code, we are plotting the distribution of 'Age' from our dataset. The function hist() is called on the 'Age' column, where 'bins' represent the number of intervals or ranges we want to use for grouping the ages. After setting a title for the plot, plt.show() displays the histogram.

Imagine you have a jar filled with different candies of various colors. If you want to count how many of each color there are, you could sort them into groups (bins) based on their colors and then create a chart showing how many of each color you have. The histogram does something similar with age data.

import seaborn as sns
sns.boxplot(x=df['Salary'])
plt.title("Salary Boxplot")
plt.show()

A box plot (or whisker plot) visually depicts groups of numerical data through their quartiles. The box represents the interquartile range where the middle 50% of the data lies, while the lines extend to show the rest of the distribution, and any outliers are shown as individual points. The sns.boxplot() function is used here to visualize the 'Salary' data, allowing us to quickly observe the central tendency and detect any anomalies (outliers) in the dataset.

Think of a box plot as a snapshot of a classroom's test scores. The box shows the range of scores where most students scored (middle 50%), while any students who scored unusually low or high (like one student scoring dramatically higher) are shown as dots outside the box.

sns.scatterplot(x='Age', y='Salary', data=df)

A scatter plot is a graph that uses dots to represent the values obtained for two different variables - one plotted along the x-axis and the other plotted along the y-axis. In this case, 'Age' is represented on the x-axis and 'Salary' on the y-axis. This helps us visualize the relationship between age and salary, identifying trends, correlations, or clusters of data points.

Think of plotting points on a map where each point represents a person’s age and salary. By looking at the pattern of points, you can often determine if older individuals tend to have higher salaries, or if the data is more scattered and less related.

sns.pairplot(df[['Age', 'Salary', 'Experience']])

A pair plot is a great way to visualize relationships between multiple variables in a dataset. It plots pairwise relationships in a grid format, providing scatter plots for numerical variables and histograms or density plots for their distributions. In this example, we are looking at 'Age', 'Salary', and 'Experience' together, revealing how these variables might relate to each other across all combinations.

Imagine viewing multiple camera angles of a conversation around a table. Each angle (or plot) provides a different perspective on how the individuals (variables) are interacting—for instance, how age might correlate with experience and salary.

sns.heatmap(df.corr(), annot=True, cmap='coolwarm')

A correlation heatmap is a graphical representation of the correlation matrix showing the relationship between variables. It uses colors to indicate the strength and direction of correlations: blue might indicate a strong negative correlation, while red indicates a strong positive correlation. The annot=True parameter adds the numeric correlation values directly onto the heatmap for clarity. This tool helps quickly identify which variables are positively or negatively correlated.

Think of a heatmap like a weather map showing temperature patterns across different regions. Just as certain areas might be hotter or cooler than others, variables in our data may have stronger or weaker relationships, with color helping us immediately see which ones need our attention.

--