Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Python – Loại bỏ phần tử ở khoảng cách K từ N


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ả.