VOOZH about

URL: https://www.geeksforgeeks.org/machine-learning/kmeans-clustering-and-pca-on-wine-dataset/

⇱ KMeans Clustering and PCA on Wine Dataset - GeeksforGeeks


  • Courses
  • Tutorials
  • Interview Prep

KMeans Clustering and PCA on Wine Dataset

Last Updated : 23 Jul, 2025

K-Means Clustering: K Means Clustering is an unsupervised learning algorithm that tries to cluster data based on their similarity. Unsupervised learning means that there is no outcome to be predicted, and the algorithm just tries to find patterns in the data. In k means clustering, we specify the number of clusters we want the data to be grouped into. The algorithm randomly assigns each observation to a set and finds the centroid of each set. Then, the algorithm iterates through two steps: Reassign data points to the cluster whose centroid is closest. Calculate the new centroid of each cluster. These two steps are repeated until the within-cluster variation cannot be reduced further. The within-cluster deviation is calculated as the sum of the Euclidean distance between the data points and their respective cluster centroids.

In this article, we will cluster the wine datasets and visualize them after dimensionality reductions with PCA.

Importing libraries needed for dataset analysis

We will first import some useful Python libraries like Pandas, Seaborn, Matplotlib and SKlearn for performing complex computational tasks. 

 Importing Dataset:

These data are the results of a chemical analysis of wines grown in the same region in Italy but derived from three different cultivars. The analysis determined the quantities of 13 constituents found in each of the three types of wines

Output :

 alcoholmalic_acidashalcalinity_of_ashmagnesiumtotal_phenolsflavanoidsnonflavanoid_phenolsproanthocyaninscolor_intensityhueod280/od315_of_diluted_winesprolinetarget
014.231.712.4315.6127.02.803.060.282.295.641.043.921065.00
113.201.782.1411.2100.02.652.760.261.284.381.053.401050.00
213.162.362.6718.6101.02.803.240.302.815.681.033.171185.00
314.371.952.5016.8113.03.853.490.242.187.800.863.451480.00
413.242.592.8721.0118.02.802.690.391.824.321.042.93735.00

Because We are doing here the unsupervised learning. So we remove the target Customer_Segment column from our datasets.

Output:

<class 'pandas.core.frame.DataFrame'>
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

Scaling the Data:

Data is scaled using StandardScaler except for the target column(Customer_Segment), whose values must remain unchanged.

Output:

 alcoholmalic_acidashalcalinity_of_ashmagnesiumtotal_phenolsflavanoidsnonflavanoid_phenolsproanthocyaninscolor_intensityhueod280/od315_of_diluted_winesproline
01.518613-0.5622500.232053-1.1695931.9139050.8089971.034819-0.6595631.2248840.2517170.3621771.8479201.013009
10.246290-0.499413-0.827996-2.4908470.0181450.5686480.733629-0.820719-0.544721-0.2933210.4060511.1134490.965242

In general, K-Means requires unlabeled data in order to run. 

So, taking data without labels to perform K-means clustering.

Elbow Method

The elbow Method is used to determine the number of clusters

Output:

👁 Elbow Curve -Geeksforgeeks
Elbow Curve

As we can see from the above graph that there is turning like an elbow at k=3. So, we can say that the right number of cluster for the given datasets is 3.

Implementing K-Means:

Let's perform the K-Means clustering for n_clusters=3.

Output :

KMeans(n_clusters=3)

For each cluster, there are values of cluster centers according to the number of columns present in the data.

Output :

array([[ 0.83523208, -0.30380968, 0.36470604, -0.61019129, 0.5775868 ,
 0.88523736, 0.97781956, -0.56208965, 0.58028658, 0.17106348,
 0.47398365, 0.77924711, 1.12518529],
 [-0.92607185, -0.39404154, -0.49451676, 0.17060184, -0.49171185,
 -0.07598265, 0.02081257, -0.03353357, 0.0582655 , -0.90191402,
 0.46180361, 0.27076419, -0.75384618],
 [ 0.16490746, 0.87154706, 0.18689833, 0.52436746, -0.07547277,
 -0.97933029, -1.21524764, 0.72606354, -0.77970639, 0.94153874,
 -1.16478865, -1.29241163, -0.40708796]])

labels_ Index of the cluster each sample belongs to.

Output :

array([0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 2, 1, 1, 1, 1,
 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 2, 1, 1, 1, 1,
 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
 1, 1, 1, 1, 1, 1, 1, 1, 2, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 2, 2,
 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
 2, 2], dtype=int32)

Apply Dimensionality Reduction Technique PCA

Principal Component Analysis is a technique that transforms high-dimensions data into lower-dimension while retaining as much information as possible.

  • It is used to interpret and visualize data. 
  • The number of variables decreases, which simplifies further analysis.

We can then view the PCA components_, i.e. the principal axes in the feature space, which represent the directions of maximum variance in the dataset. These components are sorted by explained_variance_.

Minimize the dataset from 15 features to 2 features using principal component analysis (PCA).

Output :

 PCA1PCA2
03.316751-1.443463
12.2094650.333393
22.516740-1.031151
33.757066-2.756372
41.008908-0.869831

Reducing centers:

Reducing the cluster centers using PCA.

Output :

array([[ 2.2761936 , -0.93205403],
 [-0.03695661, 1.77223945],
 [-2.72003575, -1.12565126]])

Represent the cluster plot based on PCA1 and PCA2. Differentiate clusters by passing a color parameter as c=kmeans.labels_

Output :

👁 Wine Cluster
Wine Cluster

Effect of PCA1 & PCA2 on Clusters:

If we really want to reduce the size of the dataset, the best number of principal components is much less than the number of variables in the original dataset.

Output :

array([[ 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]])

Represent the effect Features on PCA components.

Output :

👁 Effect of each features on PCA 1 and PCA2 -Geeksforgeeks
Effect of each features on PCA 1 and PCA2
Comment