#Create simple scatter plot python how to#

To demonstrate these capabilities, let's import a new dataset. For example, you could change the data's color from green to red with increasing sepalWidth. Secondly, you could change the color of each data according to a fourth variable. To use the Iris dataset as an example, you could increase the size of each data point according to its petalWidth. There are two ways of doing this.įirst, you can change the size of the scatterplot bubbles according to some variable.

#Create simple scatter plot python how to#

How To Deal With More Than 2 Variables in Python Visualizations Using MatplotlibĪs a data scientist, you will often encounter situations where you need to work with more than 2 data points in a visualizations. In the next section of this article, we will learn how to visualize 3rd and 4th variables in matplotlib by using the c and s variables that we have recently been working with. legend (handles =legend_aliases, loc = 'upper center', ncol = 3 )Īs you can see, assigning different colors to different categories (in this case, species) is a useful visualization tool in matplotlib. We will go through this process step-by-step below.įirst, let's determine the unique values of the species variable that we created by wrapping it in a set function:

Pass in this list of numbers to the cmap function.Create a new list of colors, where each color in the new list corresponds to a string from the old list.Determine the unique values of the species column.To create a color map, there are a few steps: Matplotlib's color map styles are divided into various categories, including:Ī list of some matplotlib color maps is below. One other important concept to understand is that matplotlib includes a number of color map styles by default.

We can apply this formatting to a scatterplot.Matplotlib allows us to map certain categories (in this case, species) to specific colors.This is a bunch of jargon that can be simplified as follows: A 2D array in which the rows are RGB or RGBA.A color map is a set of RGBA colors built into matplotlib that can be 'mapped' to specific values in a data set.Īlongside cmap, we will also need a variable c which is can take a few different forms: For this new species variable, we will use a matplotlib function called cmap to create a 'color map'.