When it is required to remove elements that are less than K difference away in a list, a simple iteration and ‘if’ condition is used.
Example
Below is a demonstration of the same −
my_list = [13, 29, 24, 18, 40, 15]
print("The list is :")
print(my_list)
K = 3
my_list = sorted(my_list)
index = 0
while index < len(my_list) - 1:
if my_list[index] + K > my_list[index + 1]:
del my_list[index + 1]
else:
index += 1
print("The result is :")
print(my_list)Output
The list is : [13, 29, 24, 18, 40, 15] The result is : [13, 18, 24, 29, 40]
Explanation
A list is defined and displayed on the console.
The value for K is defined.
An integer is assigned to 0.
The list is then sorted using the ‘sorted’ function.
The list is iterated over, and elements which has a difference less than K are removed from the list.
Otherwise, the index is incremented.
This is the output that is displayed on the console.