K-Means Clustering using PySpark Python Last Updated : 23 Jul, 2025 Comments Improve Suggest changes Like Article Like Report In this tutorial series, we are going to cover K-Means Clustering using Pyspark. K-means is a clustering algorithm that groups data points into K distinct clusters based on their similarity. It is an unsupervised learning technique that is widely used in data mining, machine learning, and pattern recognition. The algorithm works by iteratively assigning data points to a cluster based on their distance from the cluster's centroid and then recomputing the centroid of each cluster. The process continues until the clusters' centroids converge or a maximum number of iterations is reached. K-means is simple, efficient, and effective in finding the optimal clusters for a given dataset, making it a popular choice for various applications. So, a typical clustering problem looks like this: Cluster Similar DocumentsCluster Customers based on FeaturesIdentify similar physical groupsMarket Segmentation We'll be working with a real data set about seeds, from the UCI repository: https://archive.ics.uci.edu/dataset/236/seeds Task: We have seven geometrical parameters of wheat kernels and we have to group them into three different varieties of wheat: Kama, Rosa, and Canadian. Step 1: Starting the PySpark serverĀ Python3 from pyspark.sql import SparkSession spark = SparkSession.builder.appName('cluster').getOrCreate() print('Spark Version: {}'.format(spark.version)) Output: Spark Version: 3.3.1Step 2: Load the dataset Python3 #Loading the data dataset = spark.read.csv("seeds_dataset.csv",header=True,inferSchema=True) #show the data in the above file using the below command dataset.show(5) Output: +-----+---------+-----------+-----------------+---------------+---------------------+-----------------------+ | Area|Perimeter|Compactness|Length_of_ kernel|Width_of_kernel|Asymmetry_coefficient|Length_of_kernel_groove| +-----+---------+-----------+-----------------+---------------+---------------------+-----------------------+ |15.26| 14.84| 0.871| 5.763| 3.312| 2.221| 5.22| |14.88| 14.57| 0.8811| 5.554| 3.333| 1.018| 4.956| |14.29| 14.09| 0.905| 5.291| 3.337| 2.699| 4.825| |13.84| 13.94| 0.8955| 5.324| 3.379| 2.259| 4.805| |16.14| 14.99| 0.9034| 5.658| 3.562| 1.355| 5.175| +-----+---------+-----------+-----------------+---------------+---------------------+-----------------------+ only showing top 5 rowsPrint schema Python3 #Print schema dataset.printSchema() Output: root |-- Area: double (nullable = true) |-- Perimeter: double (nullable = true) |-- Compactness: double (nullable = true) |-- Length_of_ kernel: double (nullable = true) |-- Width_of_kernel: double (nullable = true) |-- Asymmetry_coefficient: double (nullable = true) |-- Length_of_kernel_groove: double (nullable = true)Step 3: Ā Format the data using Vector Assembler into vectors which will be used as "features" Python3 from pyspark.ml.feature import VectorAssembler vec_assembler = VectorAssembler(inputCols = dataset.columns, outputCol='features') final_data = vec_assembler.transform(dataset) final_data.select('features').show(5) Output: +--------------------+ | features| +--------------------+ |[15.26,14.84,0.87...| |[14.88,14.57,0.88...| |[14.29,14.09,0.90...| |[13.84,13.94,0.89...| |[16.14,14.99,0.90...| +--------------------+ only showing top 5 rowsStep 4: Scaling the data It is a good idea to scale our data to deal with the curse of dimensionality. Python3 from pyspark.ml.feature import StandardScaler scaler = StandardScaler(inputCol="features", outputCol="scaledFeatures", withStd=True, withMean=False) # Compute summary statistics by fitting the StandardScaler scalerModel = scaler.fit(final_data) # Normalize each feature to have unit standard deviation. final_data = scalerModel.transform(final_data) final_data.select('scaledFeatures').show(5) Output: +--------------------+ | scaledFeatures| +--------------------+ |[5.24452795332028...| |[5.11393027165175...| |[4.91116018695588...| |[4.75650503761158...| |[5.54696468981581...| +--------------------+ only showing top 5 rowsStep 5: Find the number of clusters using Silhouette Score Python3 #Importing the model from pyspark.ml.clustering import KMeans from pyspark.ml.evaluation import ClusteringEvaluator silhouette_score=[] evaluator = ClusteringEvaluator(predictionCol='prediction', featuresCol='scaledFeatures', \ metricName='silhouette', distanceMeasure='squaredEuclidean') for i in range(2,10): kmeans=KMeans(featuresCol='scaledFeatures', k=i) model=kmeans.fit(final_data) predictions=model.transform(final_data) score=evaluator.evaluate(predictions) silhouette_score.append(score) print('Silhouette Score for k =',i,'is',score) Output: Silhouette Score for k = 2 is 0.6650046039315017 Silhouette Score for k = 3 is 0.5928460025426588 Silhouette Score for k = 4 is 0.44804230341047074 Silhouette Score for k = 5 is 0.47760014315974747 Silhouette Score for k = 6 is 0.42900353119793194 Silhouette Score for k = 7 is 0.4419918246535933 Silhouette Score for k = 8 is 0.395868387829853 Silhouette Score for k = 9 is 0.40541652397305605Plot the Silhouette Score graph Python3 #Visualizing the silhouette scores in a plot import matplotlib.pyplot as plt plt.plot(range(2,10),silhouette_score) plt.xlabel('k') plt.ylabel('silhouette score') plt.title('Silhouette Score') plt.show() Output: Silhouette Score Ā Since there is no definitive answer as to what value of K is an acceptable value.Ā I wantĀ to move forward with k = 3 Where a local maximum of Silhouette Score is detected. Step 6: Train the Model Python3 # Trains a k-means model. kmeans = KMeans(featuresCol='scaledFeatures',k=3) model = kmeans.fit(final_data) predictions = model.transform(final_data) Print cluster centers Python3 # Printing cluster centers centers = model.clusterCenters() print("Cluster Centers: ") for center in centers: print(center) Output: Cluster Centers: [ 4.96198582 10.97871333 37.30930808 12.44647267 8.62880781 1.80062386 10.41913733] [ 6.35645488 12.40730852 37.41990178 13.93860446 9.7892399 2.41585309 12.29286107] [ 4.07497225 10.14410142 35.89816849 11.80812742 7.54416916 3.15411286 10.38031464]Showing the result of groupings: Python3 predictions.select('prediction').show(5) Output: +----------+ |prediction| +----------+ | 0| | 0| | 0| | 0| | 0| +----------+ only showing top 5 rowsEnd Session Python3 #End Session spark.stop() Ā Comment More infoAdvertise with us Next Article Introduction to Machine Learning S sachin7221 Follow Improve Article Tags : Machine Learning AI-ML-DS ML-Clustering python Python-Pyspark +1 More Practice Tags : Machine Learningpython Similar Reads Machine Learning Tutorial Machine learning is a branch of Artificial Intelligence that focuses on developing models and algorithms that let computers learn from data without being explicitly programmed for every task. In simple words, ML teaches the systems to think and understand like humans by learning from the data.Do you 5 min read Introduction to Machine LearningIntroduction to Machine LearningMachine learning (ML) allows computers to learn and make decisions without being explicitly programmed. It involves feeding data into algorithms to identify patterns and make predictions on new data. It is used in various applications like image recognition, speech processing, language translation, 8 min read Types of Machine LearningMachine learning is the branch of Artificial Intelligence that focuses on developing models and algorithms that let computers learn from data and improve from previous experience without being explicitly programmed for every task.In simple words, ML teaches the systems to think and understand like h 13 min read What is Machine Learning Pipeline?In artificial intelligence, developing a successful machine learning model involves more than selecting the best algorithm; it requires effective data management, training, and deployment in an organized manner. A machine learning pipeline becomes crucial in this situation. A machine learning pipeli 7 min read Applications of Machine LearningMachine Learning (ML) is one of the most significant advancements in the field of technology. It gives machines the ability to learn from data and improve over time without being explicitly programmed. ML models identify patterns from data and use them to make predictions or decisions.Organizations 3 min read Python for Machine LearningMachine Learning with Python TutorialPython language is widely used in Machine Learning because it provides libraries like NumPy, Pandas, Scikit-learn, TensorFlow, and Keras. These libraries offer tools and functions essential for data manipulation, analysis, and building machine learning models. It is well-known for its readability an 5 min read Pandas TutorialPandas is an open-source software library designed for data manipulation and analysis. It provides data structures like series and DataFrames to easily clean, transform and analyze large datasets and integrates with other Python libraries, such as NumPy and Matplotlib. It offers functions for data t 6 min read NumPy Tutorial - Python LibraryNumPy (short for Numerical Python ) is one of the most fundamental libraries in Python for scientific computing. It provides support for large, multi-dimensional arrays and matrices along with a collection of mathematical functions to operate on arrays.At its core it introduces the ndarray (n-dimens 3 min read Scikit Learn TutorialScikit-learn (also known as sklearn) is a widely-used open-source Python library for machine learning. It builds on other scientific libraries like NumPy, SciPy and Matplotlib to provide efficient tools for predictive data analysis and data mining.It offers a consistent and simple interface for a ra 3 min read ML | Data Preprocessing in PythonData preprocessing is a important step in the data science transforming raw data into a clean structured format for analysis. It involves tasks like handling missing values, normalizing data and encoding variables. Mastering preprocessing in Python ensures reliable insights for accurate predictions 6 min read EDA - Exploratory Data Analysis in PythonExploratory Data Analysis (EDA) is a important step in data analysis which focuses on understanding patterns, trends and relationships through statistical tools and visualizations. Python offers various libraries like pandas, numPy, matplotlib, seaborn and plotly which enables effective exploration 6 min read Feature EngineeringWhat is Feature Engineering?Feature engineering is the process of turning raw data into useful features that help improve the performance of machine learning models. It includes choosing, creating and adjusting data attributes to make the modelâs predictions more accurate. The goal is to make the model better by providing rele 5 min read Introduction to Dimensionality ReductionWhen working with machine learning models, datasets with too many features can cause issues like slow computation and overfitting. Dimensionality reduction helps to reduce the number of features while retaining key information. Techniques like principal component analysis (PCA), singular value decom 4 min read Feature Selection Techniques in Machine LearningIn data science many times we encounter vast of features present in a dataset. But it is not necessary all features contribute equally in prediction that's where feature selection comes. It involves selecting a subset of relevant features from the original feature set to reduce the feature space whi 5 min read Feature Engineering: Scaling, Normalization, and StandardizationFeature Scaling is a technique to standardize the independent features present in the data. It is performed during the data pre-processing to handle highly varying values. If feature scaling is not done then machine learning algorithm tends to use greater values as higher and consider smaller values 6 min read Supervised LearningSupervised Machine LearningSupervised machine learning is a fundamental approach for machine learning and artificial intelligence. It involves training a model using labeled data, where each input comes with a corresponding correct output. The process is like a teacher guiding a studentâhence the term "supervised" learning. I 12 min read Linear Regression in Machine learningLinear regression is a type of supervised machine-learning algorithm that learns from the labelled datasets and maps the data points with most optimized linear functions which can be used for prediction on new datasets. It assumes that there is a linear relationship between the input and output, mea 15+ min read Logistic Regression in Machine LearningLogistic Regression is a supervised machine learning algorithm used for classification problems. Unlike linear regression which predicts continuous values it predicts the probability that an input belongs to a specific class. It is used for binary classification where the output can be one of two po 11 min read Decision Tree in Machine LearningA decision tree is a supervised learning algorithm used for both classification and regression tasks. It has a hierarchical tree structure which consists of a root node, branches, internal nodes and leaf nodes. It It works like a flowchart help to make decisions step by step where: Internal nodes re 9 min read Random Forest Algorithm in Machine LearningRandom Forest is a machine learning algorithm that uses many decision trees to make better predictions. Each tree looks at different random parts of the data and their results are combined by voting for classification or averaging for regression. This helps in improving accuracy and reducing errors. 5 min read K-Nearest Neighbor(KNN) AlgorithmK-Nearest Neighbors (KNN) is a supervised machine learning algorithm generally used for classification but can also be used for regression tasks. It works by finding the "k" closest data points (neighbors) to a given input and makesa predictions based on the majority class (for classification) or th 8 min read Support Vector Machine (SVM) AlgorithmSupport Vector Machine (SVM) is a supervised machine learning algorithm used for classification and regression tasks. It tries to find the best boundary known as hyperplane that separates different classes in the data. It is useful when you want to do binary classification like spam vs. not spam or 9 min read Naive Bayes ClassifiersNaive Bayes is a classification algorithm that uses probability to predict which category a data point belongs to, assuming that all features are unrelated. This article will give you an overview as well as more advanced use and implementation of Naive Bayes in machine learning. Illustration behind 7 min read Unsupervised LearningWhat is Unsupervised Learning?Unsupervised learning is a branch of machine learning that deals with unlabeled data. Unlike supervised learning, where the data is labeled with a specific category or outcome, unsupervised learning algorithms are tasked with finding patterns and relationships within the data without any prior knowl 8 min read K means Clustering â IntroductionK-Means Clustering is an Unsupervised Machine Learning algorithm which groups unlabeled dataset into different clusters. It is used to organize data into groups based on their similarity. Understanding K-means ClusteringFor example online store uses K-Means to group customers based on purchase frequ 4 min read Hierarchical Clustering in Machine LearningHierarchical clustering is used to group similar data points together based on their similarity creating a hierarchy or tree-like structure. The key idea is to begin with each data point as its own separate cluster and then progressively merge or split them based on their similarity. Lets understand 7 min read DBSCAN Clustering in ML - Density based clusteringDBSCAN is a density-based clustering algorithm that groups data points that are closely packed together and marks outliers as noise based on their density in the feature space. It identifies clusters as dense regions in the data space separated by areas of lower density. Unlike K-Means or hierarchic 6 min read Apriori AlgorithmApriori Algorithm is a basic method used in data analysis to find groups of items that often appear together in large sets of data. It helps to discover useful patterns or rules about how items are related which is particularly valuable in market basket analysis. Like in a grocery store if many cust 6 min read Frequent Pattern Growth AlgorithmThe FP-Growth (Frequent Pattern Growth) algorithm efficiently mines frequent itemsets from large transactional datasets. Unlike the Apriori algorithm which suffers from high computational cost due to candidate generation and multiple database scans. FP-Growth avoids these inefficiencies by compressi 5 min read ECLAT Algorithm - MLECLAT stands for Equivalence Class Clustering and bottom-up Lattice Traversal. It is a data mining algorithm used to find frequent itemsets in a dataset. These frequent itemsets are then used to create association rules which helps to identify patterns in data. It is an improved alternative to the A 3 min read Principal Component Analysis(PCA)PCA (Principal Component Analysis) is a dimensionality reduction technique used in data analysis and machine learning. It helps you to reduce the number of features in a dataset while keeping the most important information. It changes your original features into new features these new features donât 7 min read Model Evaluation and TuningEvaluation Metrics in Machine LearningWhen building machine learning models, itâs important to understand how well they perform. Evaluation metrics help us to measure the effectiveness of our models. Whether we are solving a classification problem, predicting continuous values or clustering data, selecting the right evaluation metric al 9 min read Regularization in Machine LearningRegularization is an important technique in machine learning that helps to improve model accuracy by preventing overfitting which happens when a model learns the training data too well including noise and outliers and perform poor on new data. By adding a penalty for complexity it helps simpler mode 7 min read Cross Validation in Machine LearningCross-validation is a technique used to check how well a machine learning model performs on unseen data. It splits the data into several parts, trains the model on some parts and tests it on the remaining part repeating this process multiple times. Finally the results from each validation step are a 7 min read Hyperparameter TuningHyperparameter tuning is the process of selecting the optimal values for a machine learning model's hyperparameters. These are typically set before the actual training process begins and control aspects of the learning process itself. They influence the model's performance its complexity and how fas 7 min read ML | Underfitting and OverfittingMachine learning models aim to perform well on both training data and new, unseen data and is considered "good" if:It learns patterns effectively from the training data.It generalizes well to new, unseen data.It avoids memorizing the training data (overfitting) or failing to capture relevant pattern 5 min read Bias and Variance in Machine LearningThere are various ways to evaluate a machine-learning model. We can use MSE (Mean Squared Error) for Regression; Precision, Recall, and ROC (Receiver operating characteristics) for a Classification Problem along with Absolute Error. In a similar way, Bias and Variance help us in parameter tuning and 10 min read Advance Machine Learning TechniqueReinforcement LearningReinforcement Learning (RL) is a branch of machine learning that focuses on how agents can learn to make decisions through trial and error to maximize cumulative rewards. RL allows machines to learn by interacting with an environment and receiving feedback based on their actions. This feedback comes 6 min read Semi-Supervised Learning in MLToday's Machine Learning algorithms can be broadly classified into three categories, Supervised Learning, Unsupervised Learning, and Reinforcement Learning. Casting Reinforced Learning aside, the primary two categories of Machine Learning problems are Supervised and Unsupervised Learning. The basic 4 min read Self-Supervised Learning (SSL)In this article, we will learn a major type of machine learning model which is Self-Supervised Learning Algorithms. Usage of these algorithms has increased widely in the past times as the sizes of the model have increased up to billions of parameters and hence require a huge corpus of data to train 8 min read Ensemble LearningEnsemble learning is a method where we use many small models instead of just one. Each of these models may not be very strong on its own, but when we put their results together, we get a better and more accurate answer. It's like asking a group of people for advice instead of just one personâeach on 8 min read Machine Learning PracticeTop 50+ Machine Learning Interview Questions and AnswersMachine Learning involves the development of algorithms and statistical models that enable computers to improve their performance in tasks through experience. Machine Learning is one of the booming careers in the present-day scenario.If you are preparing for machine learning interview, this intervie 15+ min read 100+ Machine Learning Projects with Source Code [2025]This article provides over 100 Machine Learning projects and ideas to provide hands-on experience for both beginners and professionals. Whether you're a student enhancing your resume or a professional advancing your career these projects offer practical insights into the world of Machine Learning an 5 min read Like