Tutorial: K Nearest Neighbors in Python
In this post, we’ll be using the Knearest neighbors algorithm to predict how many points NBA players scored in the 20132014 season. Along the way, we’ll learn about euclidean distance and figure out which NBA players are the most similar to Lebron James. If you want to follow along, you can grab the dataset in csv format here.
A look at the data
Before we dive into the algorithm, let’s take a look at our data. Each row in the data contains information on how a player performed in the 20132014 NBA season.
Here are some selected columns from the data:

player
— name of the player 
pos
— the position of the player 
g
— number of games the player was in 
gs
— number of games the player started 
pts
— total points the player scored
There are many more columns in the data, mostly containing information about average player game performance over the course of the season. See this site for an explanation of the rest of them.
We can read our dataset in and figure out which columns are present:
import pandas
with open("nba_2013.csv", 'r') as csvfile:
nba = pandas.read_csv(csvfile)
print(nba.columns.values) # The names of all the columns in the data.
['player' 'pos' 'age' 'bref_team_id' 'g' 'gs' 'mp' 'fg' 'fga' 'fg.' 'x3p' 'x3pa' 'x3p.' 'x2p' 'x2pa' 'x2p.' 'efg.' 'ft' 'fta' 'ft.' 'orb' 'drb' 'trb' 'ast' 'stl' 'blk' 'tov' 'pf' 'pts' 'season' 'season_end']
KNN overview
The knearest neighbors algorithm is based around the simple idea of predicting unknown values by matching them with the most similar known values.
Let’s say that we have 3 different types of cars. We know the name of the car, its horsepower, whether or not it has racing stripes, and whether or not it’s fast.:
car,horsepower,racing_stripes,is_fast
Honda Accord,180,False,False
Yugo,500,True,True
Delorean DMC12,200,True,True
Let’s say that we now have another car, but we don’t know how fast it is:
car,horsepower,racing_stripes,is_fast
Chevrolet Camaro,400,True,Unknown
We want to figure out if the car is fast or not. In order to predict if it is with k nearest neighbors, we first find the most similar known car. In this case, we would compare the horsepower
and racing_stripes
values to find the most similar car, which is the Yugo
. Since the Yugo is fast, we would predict that the Camaro is also fast. This is an example of 1nearest neighbors — we only looked at the most similar car, giving us a k of 1.
If we performed a 2nearest neighbors, we would end up with 2 True
values (for the Delorean and the Yugo), which would average out to True
. The Delorean and Yugo are the two most similar cars, giving us a k of 2.
If we did 3nearest neighbors, we would end up with 2 True
values and a False
value, which would average out to True
.
The number of neighbors we use for knearest neighbors (k) can be any value less than the number of rows in our dataset. In practice, looking at only a few neighbors makes the algorithm perform better, because the less similar the neighbors are to our data, the worse the prediction will be.
Euclidean distance
Before we can predict using KNN, we need to find some way to figure out which data rows are “closest” to the row we’re trying to predict on.
A simple way to do this is to use Euclidean distance. The formula is \(\sqrt{(q_1p_1)^2 + (q_2p_2)^2 + \cdots + (q_np_n)^2}\)
Let’s say we have these two rows (True/False has been converted to 1/0), and we want to find the distance between them:
car,horsepower,is_fast
Honda Accord,180,0
Chevrolet Camaro,400,1
We would first only select the numeric columns. Then the distance becomes \(\sqrt{(180400)^2 + (01)^2}\), which is about equal to 220
.
We can use the principle of euclidean distance to find the most similar NBA players to Lebron James.
# Select Lebron James from our dataset
selected_player = nba[nba["player"] == "LeBron James"].iloc[0]
# Choose only the numeric columns (we'll use these to compute euclidean distance)
distance_columns = ['age', 'g', 'gs', 'mp', 'fg', 'fga', 'fg.', 'x3p', 'x3pa', 'x3p.', 'x2p', 'x2pa', 'x2p.', 'efg.', 'ft', 'fta', 'ft.', 'orb', 'drb', 'trb', 'ast', 'stl', 'blk', 'tov', 'pf', 'pts']
def euclidean_distance(row):
"""
A simple euclidean distance function
"""
inner_value = 0
for k in distance_columns:
inner_value += (row[k]  selected_player[k]) ** 2
return math.sqrt(inner_value)
# Find the distance from each player in the dataset to lebron.
lebron_distance = nba.apply(euclidean_distance, axis=1)
Normalizing columns
You may have noticed that horsepower
in the cars example had a much larger impact on the final distance than racing_stripes
did. This is because horsepower
values are much larger in absolute terms, and thus dwarf the impact of racing_stripes
values in the euclidean distance calculations.
This can be bad, because a variable having larger values doesn’t necessarily make it better at predicting what rows are similar.
A simple way to deal with this is to normalize all the columns to have a mean of 0, and a standard deviation of 1. This will ensure that no single column has a dominant impact on the euclidean distance calculations.
To set the mean to 0, we have to find the mean of a column, then subtract the mean from every value in the column. To set the standard deviation to 1, we divide every value in the column by the standard deviation. The formula is \(x=\frac{x\mu}{\sigma}\).
# Select only the numeric columns from the NBA dataset
nba_numeric = nba[distance_columns]
# Normalize all of the numeric columns
nba_normalized = (nba_numeric  nba_numeric.mean()) / nba_numeric.std()
Finding the nearest neighbor
We now know enough to find the nearest neighbor of a given row in the NBA dataset. We can use the distance.euclidean
function from scipy.spatial
, a much faster way to calculate euclidean distance.
from scipy.spatial import distance
# Fill in NA values in nba_normalized
nba_normalized.fillna(0, inplace=True)
# Find the normalized vector for lebron james.
lebron_normalized = nba_normalized[nba["player"] == "LeBron James"]
# Find the distance between lebron james and everyone else.
euclidean_distances = nba_normalized.apply(lambda row: distance.euclidean(row, lebron_normalized), axis=1)
# Create a new dataframe with distances.
distance_frame = pandas.DataFrame(data={"dist": euclidean_distances, "idx": euclidean_distances.index})
distance_frame.sort("dist", inplace=True)
# Find the most similar player to lebron (the lowest distance to lebron is lebron, the second smallest is the most similar nonlebron player)
second_smallest = distance_frame.iloc[1]["idx"]
most_similar_to_lebron = nba.loc[int(second_smallest)]["player"]
Generating training and testing sets
Now that we know how to find the nearest neighbors, we can make predictions on a test set. We’ll try to predict how many points a player scored using the 5
closest neighbors. We’ll find neighbors by using all the numeric columns in the dataset to generate similarity scores.
First, we have to generate test and train sets. In order to do this, we’ll use random sampling. We’ll randomly shuffle the index of the nba
dataframe, and then pick rows using the randomly shuffled values.
If we didn’t do this, we’d end up predicting and training on the same data set, which would overfit. We could do cross validation also, which would be slightly better, but slightly more complex.
import random
from numpy.random import permutation
# Randomly shuffle the index of nba.
random_indices = permutation(nba.index)
# Set a cutoff for how many items we want in the test set (in this case 1/3 of the items)
test_cutoff = math.floor(len(nba)/3)
# Generate the test set by taking the first 1/3 of the randomly shuffled indices.
test = nba.loc[random_indices[1:test_cutoff]]
# Generate the train set with the rest of the data.
train = nba.loc[random_indices[test_cutoff:]]
Using sklearn for k nearest neighbors
Instead of having to do it all ourselves, we can use the knearest neighbors implementation in scikitlearn. Here’s the documentation. There’s a regressor and a classifier available, but we’ll be using the regressor, as we have continuous values to predict on.
Sklearn performs the normalization and distance finding automatically, and lets us specify how many neighbors we want to look at.
# The columns that we will be making predictions with.
x_columns = ['age', 'g', 'gs', 'mp', 'fg', 'fga', 'fg.', 'x3p', 'x3pa', 'x3p.', 'x2p', 'x2pa', 'x2p.', 'efg.', 'ft', 'fta', 'ft.', 'orb', 'drb', 'trb', 'ast', 'stl', 'blk', 'tov', 'pf']
# The column that we want to predict.
y_column = ["pts"]
from sklearn.neighbors import KNeighborsRegressor
# Create the knn model.
# Look at the five closest neighbors.
knn = KNeighborsRegressor(n_neighbors=5)
# Fit the model on the training data.
knn.fit(train[x_columns], train[y_column])
# Make point predictions on the test set using the fit model.
predictions = knn.predict(test[x_columns])
Computing error
Now that we know our point predictions, we can compute the error involved with our predictions. We can compute mean squared error. The formula is \(\frac{1}{n}\sum_{i=1}^{n}(\hat{y_{i}} – y_{i})^{2}\).
# Get the actual values for the test set.
actual = test[y_column]
# Compute the mean squared error of our predictions.
mse = (((predictions  actual) ** 2).sum()) / len(predictions)
Next Steps
For more on k nearest neighbors, you can check out our sixpart interactive machine learning fundamentals course, which teaches the basics of machine learning using the k nearest neighbors algorithm.
Vik is the CEO and Founder of Dataquest.