Khi cần loại bỏ các phần tử nằm trong khoảng [N-K, N+K] (tức là phần tử có khoảng cách đến N không vượt quá K), ta dùng list comprehension với điều kiện.
Ví dụ
my_list = [13, 52, 5, 45, 65, 61, 18]
print("Danh sách ban đầu:")
print(my_list)
K = 3
print("Giá trị K:")
print(K)
N = 5
print("Giá trị N:")
print(N)
my_result = [element for element in my_list if element < N - K or element > N + K]
print("Kết quả:")
print(my_result)
Kết quả
Danh sách ban đầu: [13, 52, 5, 45, 65, 61, 18] Giá trị K: 3 Giá trị N: 5 Kết quả: [13, 52, 45, 65, 61, 18]
Giải thích
Định nghĩa danh sách, K và N.
Khoảng cần loại: [N-K, N+K] = [2, 8]. Phần tử 5 nằm trong khoảng này nên bị loại.
List comprehension giữ phần tử < N-K hoặc > N+K.
In kết quả.