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

Chương trình Python in hàng có độ dài xác định từ ma trận

Khi cần in các hàng có độ dài bằng giá trị cho trước từ ma trận, ta dùng list comprehension.

Ví dụ

my_list = [[22, 4, 63, 7], [24, 4, 85], [95], [2, 55, 4, 7, 91], [5, 31, 1]]
print("Danh sách là:")
print(my_list)

my_key = 4
my_result = [sub for sub in my_list if len(sub) == my_key]

print("Danh sách kết quả:")
print(my_result)

Kết quả đầu ra

Danh sách là:
[[22, 4, 63, 7], [24, 4, 85], [95], [2, 55, 4, 7, 91], [5, 31, 1]]
Danh sách kết quả:
[[22, 4, 63, 7]]

Giải thích

  • Định nghĩa ma trận (danh sách lồng nhau) và giá trị K=4.
  • List comprehension lọc các hàng con có len(sub) == my_key.
  • Chỉ hàng đầu tiên [22, 4, 63, 7] có độ dài 4.

Mở rộng: Dùng len(sub) >= my_key cho "ít nhất K", len(sub) <= my_key cho "tối đa K".