How to Color Scatter Plots by Column Values in Python with pandas and Matplotlib?

Color Scatter Plots by Column Values in Python with pandas and Matplotlib

Introduction

As you mentioned, ggplot2 offers convenient aesthetics customization, allowing you to color scatter plots based on column values. This article explores equivalent functionalities in Python using pandas and Matplotlib.

Solution Using Seaborn

Seaborn, a data visualization library for Python, provides an elegant solution to this problem.

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

# Load and clean the data
data = pd.read_csv('data.csv')
data['Gender'] = data['Gender'].astype('category')

# Create the scatter plot with color mapping
sns.relplot(data=data, x='Weight', y='Height', hue='Gender')</code>

This code leverages the relplot function to create a scatter plot, with the hue parameter assigning colors based on the Gender column.

Solution Using Matplotlib and Dictionary

If you prefer to use Matplotlib directly, you can create a color mapping dictionary and use it to color the points.

<code class="python">import matplotlib.pyplot as plt
import numpy as np

# Load and clean the data
data = pd.read_csv('data.csv')
data['Gender'] = data['Gender'].astype('category')

# Create a color mapping dictionary
categories = np.unique(data['Gender'])
colors = np.linspace(0, 1, len(categories))
color_dict = dict(zip(categories, colors))

# Add a 'Color' column to the DataFrame
data['Color'] = data['Gender'].map(color_dict)

# Create the scatter plot
plt.scatter(data['Weight'], data['Height'], c=data['Color'])
plt.show()</code>

In this approach, the color_dict assigns colors to each category in the Gender column. The 'Color' column is added to the DataFrame, and the c parameter in the scatter function uses this column to determine the color of each point.

Additional Customization

Both Seaborn and Matplotlib allow for further customization of the scatter plot, such as adjusting the color palette or adding a legend. Refer to their documentation for more options.

Conclusion

You can easily color scatter plots by column values in Python using either Seaborn or Matplotlib directly. Seaborn provides a convenient high-level interface, while Matplotlib offers greater control over customization. By leveraging the techniques described above, you can create informative and visually appealing scatter plots in Python.

The above is the detailed content of How to Color Scatter Plots by Column Values in Python with pandas and Matplotlib?. For more information, please follow other related articles on the PHP Chinese website!