When it is required to remove elements, which are at K distance with N, a list comprehension along with a specific condition is used.
Below is a demonstration of the same −
Example
my_list = [13, 52, 5, 45, 65, 61, 18 ]
print("The list is :")
print(my_list)
K = 3
print("The value of K is ")
print(K)
N = 5
print("The value of N is ")
print(N)
my_result = [element for element in my_list if element < N - K or element > N + K]
print("The result is:")
print(my_result)Output
The list is : [13, 52, 5, 45, 65, 61, 18] The value of K is 3 The value of N is 5 The result is: [13, 52, 45, 65, 61, 18]
Explanation
A list of integers is defined and is displayed on the console.
A value for K is defined and is displayed on the console.
A value for N is defined and is displayed on the console.
A list comprehension is used to iterate over the elements and check if an element in the list is less than difference between N and K or sum of N and K.
If yes, the element is stored in a list.
This is assigned to a variable.
This is displayed as output on the console.