How to plot categorical variables as color with Matplotlib and Pandas
Land Your First Data Science Job
A proven roadmap to prepare for $75K+ entry-level data roles. Perfect for Data Scientist ready to level up their career.
Sometimes it is useful to add an extra dimension to a plot, especially to see whether there are some clusters forming.
To do so, you will first have to transform the categories into numbers and then set the color as the category number.
Here is the example
Here is the result
More on plots
If you want to know more about how to add labels, plot different types of plots, etc... check out the other articles I wrote on the topic, just here :
Land Your First Data Science Job
A proven roadmap to prepare for $75K+ entry-level data roles. Perfect for Data Scientist ready to level up their career.
Related Articles
Continue your learning journey with these related topics
Master Data Science in Days, Not Months 🚀
Skip the theoretical rabbit holes. Get practical data science skills delivered in bite-sized lessons – Approach used by real data scientist. Not bookworms. 📚