search
Search
Publish
menu
menu search toc more_vert
Robocat
Guest 0reps
Thanks for the thanks!
close
Comments
Log in or sign up
Cancel
Post
account_circle
Profile
exit_to_app
Sign out
help Ask a question
Share on Twitter
search
keyboard_voice
close
Searching Tips
Search for a recipe: "Creating a table in MySQL"
Search for an API documentation: "@append"
Search for code: "!dataframe"
Apply a tag filter: "#python"
Useful Shortcuts
/ to open search panel
Esc to close search panel
to navigate between search results
d to clear all current filters
Enter to expand content preview
icon_star
Doc Search
icon_star
Code Search Beta
SORRY NOTHING FOUND!
mic
Start speaking...
Voice search is only supported in Safari and Chrome.
Navigate to
A
A
share
thumb_up_alt
bookmark
arrow_backShare
Twitter
Facebook

Plotting scatter plot with category in Matplotlib

Programming
chevron_right
Python
chevron_right
Matplotlib
chevron_right
Cookbooks
chevron_right
Graphs Cookbook
schedule Jul 1, 2022
Last updated
local_offer PythonMatplotlib
Tags

Basic example of plotting scatter plot with categories

To plot a scatter plot with categories or classes in Matplotlib, use the following code:

import matplotlib.pyplot as plt
import pandas as pd

labels = ['A','B','A','C']
arr_int_classes = pd.Categorical(labels).codes
scatter = plt.scatter([5,2,3,3], [1,2,4,1], c=arr_int_classes)
plt.legend(handles=scatter.legend_elements()[0], labels=arr_str_labels)
plt.show()

Here, we are first converting our string labels into numerical values using Pandas' Categorical(~) function:

arr_int_classes = pd.Categorical(labels).codes
arr_int_classes
array([0, 1, 0, 2], dtype=int8)

This results in the following plot:

Plotting scatter plot with categories using custom colors

To plot a scatter plot with categories, use ListedColormap:

import matplotlib.pyplot as plt
from matplotlib.colors import ListedColormap
import pandas as pd

labels = ['A','B','A','C']
arr_int_classes = pd.Categorical(labels).codes
colours = ListedColormap(['g','blue','#EA131B'])
scatter = plt.scatter([5,2,3,3], [1,2,4,1], c=arr_int_classes, cmap=colours)
plt.legend(handles=scatter.legend_elements()[0], labels=arr_str_labels)
plt.show()

This produces the following plot:

robocat
Published by Isshin Inada
Edited by 0 others
Did you find this page useful?
thumb_up
thumb_down
Ask a question or leave a feedback...
thumb_up
0
thumb_down
0
chat_bubble_outline
0
settings
Enjoy our search
Hit / to insta-search docs and recipes!