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