Python – Remove Elements in K distance with N

PythonServer Side ProgrammingProgramming

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

 Live Demo

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.

raja
Published on 06-Sep-2021 07:52:54
Advertisements