- Data Structure
- Networking
- RDBMS
- Operating System
- Java
- MS Excel
- iOS
- HTML
- CSS
- Android
- Python
- C Programming
- C++
- C#
- MongoDB
- MySQL
- Javascript
- PHP
- Physics
- Chemistry
- Biology
- Mathematics
- English
- Economics
- Psychology
- Social Studies
- Fashion Studies
- Legal Studies
- Selected Reading
- UPSC IAS Exams Notes
- Developer's Best Practices
- Questions and Answers
- Effective Resume Writing
- HR Interview Questions
- Computer Glossary
- Who is Who
Python – Remove Elements in K distance with N
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.
Advertisements