Computer >> บทช่วยสอนคอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

นับจำนวน Element ทั้งหมดใน Nested List ใน Python

นับ Element ทั้งหมดใน Nested List (Flatten Count)

ลิสต์ที่มีลิสต์ซ้อนอยู่ภายใน ต้องการนับจำนวน Element ทั้งหมด (Flatten แล้วนับ)

วิธีที่ 1: List Comprehension Flatten + len()


lst = [[2,9,6], [5,'a'], [0], [12,'c',9,3]]
count = len([x for sub in lst for x in sub])
print(count)  # 10

วิธีที่ 2: ใช้ itertools.chain


from itertools import chain
count = len(list(chain(*lst)))
print(count)  # 10