"Object-Oriented Swift" was retired on May 31, 2020. You are now viewing the recommended replacement.
Heads up! To view this whole video, sign in with your Courses account or enroll in your free 7-day trial. Sign In Enroll
Well done!
You have completed Introduction to Data Visualization with Matplotlib!
You have completed Introduction to Data Visualization with Matplotlib!
Preview
Let’s utilize a scatter plot to see what correlations if any, there are between the sepal length and width based on the variety of iris.
Color Dict
colors = {"Iris-setosa": "#2B5B84", "Iris-versicolor": "g", "Iris-virginica": "purple"}
Correlations
- Positive Correlation: as one variable increases so does the other. Height and shoe size are an example; as one's height increases so does the shoe size.
- Negative Correlation: as one variable increases, the other decreases. Time spent studying and time spent on video games are negatively correlated; as your time studying increases, time spent on video games decreases.
- No Correlation: there is no apparent relationship between the variables. Video game scores and shoe size appear to have no correlation; as one increases, the other one is not affected.
Further Reading
Related Discussions
Have questions about this video? Start a discussion with the community and Treehouse staff.
Sign upRelated Discussions
Have questions about this video? Start a discussion with the community and Treehouse staff.
Sign up
We left off with our data in a list,
just waiting to be used.
0:00
Let's utilize a scatter plot to
see what correlations, if any,
0:04
there are between the sepal length and
width, based on the variety of aggregates.
0:07
[SOUND] Recall that scatter plots are used
to show how much one variable is impacted
0:12
by another, or its correlation.
0:16
We use scatter plots to show
relationships between values.
0:18
In this case, our sepal length and width.
0:22
The scatter plot allows us to quickly
visualize the distribution of the data and
0:24
notice any outliers.
0:28
We can see if there’s a positive,
negative, or
0:30
nonexistent correlation between our
data based on the scatter plot results.
0:33
Let's jump back into our Python code and
develop our chart.
0:38
Let's get started from
the previous video's code and
0:42
rename name this notebook iris_scatter.
0:46
Since we'll be starting
our work with plots now,
0:58
we'll need to have our matplotlib.pyplot
import to our project.
1:02
matplotlib.pyplot as plt.
1:06
Let's create a dict for
1:12
our marker colors that we can use as
we loop through our list information.
1:13
The colors allow us to see the different
iris classes more easily, and
1:18
visualize a third
variable in our data set.
1:22
I'll paste that dict in here.
1:25
I have included a copy of
it in the teacher's notes.
1:27
Our colors then are the blue hue for
setosa.
1:38
Green, that short code, for versicolor.
1:42
And purple for virginica.
1:45
Our list of iris data also includes an
extra item that we don't need at the end,
1:48
so let's pop that off.
1:52
Now we'll want to loop through our array,
and assign our x and
1:57
y-values to the sepal length and width.
2:01
These are located in the first and
second columns of our array, respectively.
2:04
We can use a function in the itertools
library called groupby that allows us to
2:09
easily do that.
2:14
Let's add that import first and
I'll show you the code.
2:15
From itertools import groupby.
2:23
If you haven't used itertools,
it is a module that provides functions for
2:30
efficient looping.
2:34
Check the teacher's notes for
additional information.
2:36
So to start this, species and
2:38
group in groupby,
2:45
Group is a generator, so
you can only go over it one time.
2:54
And then we'll get our sepal length.
3:09
It's gonna be the float value.
3:16
Sepal widths, similar.
3:32
Then we assign that to plt.scatter.
3:51
Sepal_lengths, sepal_widths for
our y-value there.
3:55
We'll assign it a marker size of 10.
4:05
C for the colors will come from our
colors dict, and grab the species.
4:10
And we'll label based on species as well.
4:17
Now, before we call plt.show,
let's add a plot title, axes labels,
4:22
and legend to our chart here
to add context to our data.
4:27
This is an important thing to remember.
4:31
Always label your axes,
legends, and charts.
4:34
Plt.title, Fisher's Iris Data Set.
4:38
We'll give that a fontsize of 12.
4:47
Bring that up a little bit.
4:54
Our xlabel.
4:58
These are our sepal
lengths in centimeters.
5:01
We’ll assign that a fontsize of 10.
5:07
For our ylabel,
these are our sepal widths.
5:11
Again, in centimeters, and
we'll give that a fontsize of 10 as well.
5:16
We'll call plt.legend, And
5:27
we'll give this a location
in the upper right.
5:31
Here we are setting the legend location
to be displayed in the upper right
5:37
of the chart.
5:40
But we could display it in the upper left,
upper center, bottom left, etc.
5:41
Since there aren't any data points
being displayed in the upper right,
5:46
that seems like a good position.
5:50
Now we just call plt.show.
5:52
And run our cell.
5:58
We can see some patterns
here in our sepal data.
6:05
Iris-setosa is a pretty good grouping in
the upper left quadrant of our chart.
6:07
There are some outliers though.
6:12
The other two varieties seem
to be clumped together and
6:14
intermixed with some
even greater outliers.
6:17
Our plot looks a bit small here, though.
6:20
Let's assign a size to our figure
to make it a bit easier to see.
6:22
We do that, Go up here,
6:26
right under input_file,
that's kind of a standard spot for it.
6:30
We attach something to the figure object,
figsize.
6:36
7.5, 4.25 seems to work pretty well,
and we can run our cell again.
6:44
There, that's better.
6:54
From an analysis standpoint, we could draw
some conclusions based on this chart.
6:57
It appears that all three iris
varieties have a positive correlation
7:01
between sepal length and width.
7:05
Iris-setosa has a better defined positive
correlation than the other varieties.
7:06
Scatter plots are, of course,
only one way to explore our data.
7:13
You need to sign up for Treehouse in order to download course files.
Sign upYou need to sign up for Treehouse in order to set up Workspace
Sign up