# Principal Component Analysis (PCA) with Scikit-Learn

Learn how to perform principal component analysis (PCA) in Python using the scikit-learn library.

Image by Author

If you’re familiar with the unsupervised learning paradigm, you’d have come across dimensionality reduction and the algorithms used for dimensionality reduction such as the **principal component analysis** (PCA). Datasets for machine learning typically contain a large number of features, but such high-dimensional feature spaces are not always helpful.

In general, all the features are *not* equally important and there are certain features that account for a large percentage of variance in the dataset. Dimensionality reduction algorithms aim to reduce the dimension of the feature space to a fraction of the original number of dimensions. In doing so, the features with high variance are still retained—but are in the transformed feature space. And principal component analysis (PCA) is one of the most popular dimensionality reduction algorithms.

In this tutorial, we’ll learn how principal component analysis (PCA) works and how to implement it using the scikit-learn library.

# How Does Principal Component Analysis (PCA) Work?

Before we go ahead and implement principal component analysis (PCA) in scikit-learn, it’s helpful to understand how PCA works.

As mentioned, principal component analysis is a dimensionality reduction algorithm. Meaning it reduces the dimensionality of the feature space. But how does it achieve this reduction?

The motivation behind the algorithm is that there are certain features that capture a large percentage of variance in the original dataset. So it's important to find the **directions of maximum variance** in the dataset. These directions are called **principal components**. And PCA is essentially a projection of the dataset onto the principal components.

So how do we find the principal components?

Suppose the data matrix X is of dimensions **num_observations x num_features**, we perform eigenvalue decomposition on the covariance matrix of X.

If the features are all zero mean, then the covariance matrix is given by X.T X. Here, X.T is the transpose of the matrix X. If the features are not all zero mean initially, we can subtract the mean of column i from each entry in that column and compute the covariance matrix. It’s simple to see that the covariance matrix is a square matrix of order **num_features**.

Image by Author

The first k principal components are the *eigenvectors* corresponding to the *k largest eigenvalues*.

So the steps in PCA can be summarized as follows:

Image by Author

Because the covariance matrix is a symmetric and positive semi-definite, the eigendecomposition takes the following form:

X.T X = D Λ D.T

Where, D is the matrix of eigenvectors and Λ is a diagonal matrix of eigenvalues.

# Principal Components Using SVD

Another matrix factorization technique that can be used to compute principal components is singular value decomposition or SVD.

Singular value decomposition (SVD) is defined for all matrices. Given a matrix X, SVD of X gives: X = U Σ V.T. Here, U, Σ, and V are the matrices of left singular vectors, singular values, and right singular vectors, respectively. V.T. is the transpose of V.

So the SVD of the covariance matrix of X is given by:

Comparing the equivalence of the two matrix decompositions:

We have the following:

There are computationally efficient algorithms for calculating the SVD of a matrix. The scikit-learn implementation of PCA also uses SVD under the hood to compute the principal components.

# Performing Principal Component Analysis (PCA) with Scikit-Learn

Now that we’ve learned the basics of principal component analysis, let’s proceed with the scikit-learn implementation of the same.

## Step 1 – Load the Dataset

To understand how to implement principal component analysis, let’s use a simple dataset. In this tutorial, we’ll use the wine dataset available as part of scikit-learn's **datasets** module.

Let’s start by loading and preprocessing the dataset:

```
from sklearn import datasets
wine_data = datasets.load_wine(as_frame=True)
df = wine_data.data
```

It has 13 features and 178 records in all.

```
print(df.shape)
Output >> (178, 13)
```

```
print(df.info())
Output >>
```
RangeIndex: 178 entries, 0 to 177
Data columns (total 13 columns):
# Column Non-Null Count Dtype
--- ------ -------------- -----
0 alcohol 178 non-null float64
1 malic_acid 178 non-null float64
2 ash 178 non-null float64
3 alcalinity_of_ash 178 non-null float64
4 magnesium 178 non-null float64
5 total_phenols 178 non-null float64
6 flavanoids 178 non-null float64
7 nonflavanoid_phenols 178 non-null float64
8 proanthocyanins 178 non-null float64
9 color_intensity 178 non-null float64
10 hue 178 non-null float64
11 od280/od315_of_diluted_wines 178 non-null float64
12 proline 178 non-null float64
dtypes: float64(13)
memory usage: 18.2 KB
None

## Step 2 – Preprocess the Dataset

As a next step, let's preprocess the dataset. The features are all on different scales. To bring them all to a common scale, we’ll use the `StandardScaler`

that transforms the features to have zero mean and unit variance:

```
from sklearn.preprocessing import StandardScaler
std_scaler = StandardScaler()
scaled_df = std_scaler.fit_transform(df)
```

## Step 3 – Perform PCA on the Preprocessed Dataset

To find the principal components, we can use the PCA class from scikit-learn’s **decomposition** module.

Let’s instantiate a PCA object by passing in the number of principal components `n_components`

to the constructor.

The number of principal components is the number of dimensions that you’d like to reduce the feature space to. Here, we set the number of components to 3.

```
from sklearn.decomposition import PCA
pca = PCA(n_components=3)
pca.fit_transform(scaled_df)
```

Instead of calling the `fit_transform()`

method, you can also call `fit()`

followed by the `transform()`

method.

Notice how the steps in principal component analysis such as computing the covariance matrix, performing eigendecomposition or singular value decomposition on the covariance matrix to get the principal components have all been abstracted away when we use scikit-learn’s implementation of PCA.

## Step 4 – Examining Some Useful Attributes of the PCA Object

The PCA instance `pca`

that we created has several useful attributes that help us understand what is going on under the hood.

The attribute `components_`

stores the directions of maximum variance (the principal components).

`print(pca.components_)`

```
Output >>
[[ 0.1443294 -0.24518758 -0.00205106 -0.23932041 0.14199204 0.39466085
0.4229343 -0.2985331 0.31342949 -0.0886167 0.29671456 0.37616741
0.28675223]
[-0.48365155 -0.22493093 -0.31606881 0.0105905 -0.299634 -0.06503951
0.00335981 -0.02877949 -0.03930172 -0.52999567 0.27923515 0.16449619
-0.36490283]
[-0.20738262 0.08901289 0.6262239 0.61208035 0.13075693 0.14617896
0.1506819 0.17036816 0.14945431 -0.13730621 0.08522192 0.16600459
-0.12674592]]
```

We mentioned that the principal components are directions of maximum variance in the dataset. But how do we measure *how much of the total variance* is captured in the number of principal components we just chose?

The `explained_variance_ratio_`

attribute captures the ratio of the total variance each principal component captures. Sowe can sum up the ratios to get the total variance in the chosen number of components.

`print(sum(pca.explained_variance_ratio_))`

`Output >> 0.6652996889318527`

Here, we see that three principal components capture over 66.5% of total variance in the dataset.

## Step 5 – Analyzing the Change in Explained Variance Ratio

We can try running principal component analysis by varying the number of components `n_components`

.

```
import numpy as np
nums = np.arange(14)
```

```
var_ratio = []
for num in nums:
pca = PCA(n_components=num)
pca.fit(scaled_df)
var_ratio.append(np.sum(pca.explained_variance_ratio_))
```

To visualize the `explained_variance_ratio_`

for the number of components, let’s plot the two quantities as shown:

```
import matplotlib.pyplot as plt
plt.figure(figsize=(4,2),dpi=150)
plt.grid()
plt.plot(nums,var_ratio,marker='o')
plt.xlabel('n_components')
plt.ylabel('Explained variance ratio')
plt.title('n_components vs. Explained Variance Ratio')
```

When we use all the 13 components, the `explained_variance_ratio_`

is 1.0 indicating that we’ve captured 100% of the variance in the dataset.

In this example, we see that with 6 principal components, we'll be able to capture more than 80% of variance in the input dataset.

# Conclusion

I hope you’ve learned how to perform principal component analysis using built-in functionality in the scikit-learn library. Next, you can try to implement PCA on a dataset of your choice. If you’re looking for good datasets to work with, check out this list of websites to find datasets for your data science projects.

# Further Reading

[1] Computational Linear Algebra, fast.ai

**Bala Priya C** is a developer and technical writer from India. She likes working at the intersection of math, programming, data science, and content creation. Her areas of interest and expertise include DevOps, data science, and natural language processing. She enjoys reading, writing, coding, and coffee! Currently, she's working on learning and sharing her knowledge with the developer community by authoring tutorials, how-to guides, opinion pieces, and more.