พิมพ์แถวที่มีความยาวเท่ากับ K จาก Matrix

แนวคิด

จาก matrix (ลิสต์ของลิสต์) ให้คืนเฉพาะแถวที่มีจำนวนสมาชิกเท่ากับ K ที่กำหนด

ตัวอย่างโค้ด

my_list = [[22, 4, 63, 7], [24, 4, 85], [95], [2, 55, 4, 7, 91], [5, 31, 1]]
print("Matrix:", my_list)

my_key = 4
print("K =", my_key)

my_result = [sub for sub in my_list if len(sub) == my_key]
print("แถวที่ยาว 4:", my_result)

ผลลัพธ์

Matrix: [[22, 4, 63, 7], [24, 4, 85], [95], [2, 55, 4, 7, 91], [5, 31, 1]]
K = 4
แถวที่ยาว 4: [[22, 4, 63, 7]]

การขยาย

# แถวที่ยาวมากกว่า K
[sub for sub in my_list if len(sub) > my_key]

# แถวที่ยาวน้อยกว่า K
[sub for sub in my_list if len(sub) < my_key]

# แถวที่ยาวในช่วง
[sub for sub in my_list if 2 <= len(sub) <= 4]