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

Đếm tổng số phần tử trong list (có lồng nhau) trong Python


Đếm số phần tử trong list, kể cả các phần tử bên trong list lồng nhau (flatten).

Cách 1: List comprehension + len()

len([x for y in listA for x in y]) làm phẳng list rồi đếm.

listA = [[2,9,6], [5,'a'], [0], [12,'c',9,3]]
res = len([x for y in listA for x in y])
print("Tổng số phần tử:", res)  # 10

Cách 2: itertools.chain()

Dùng chain(*listA) làm phẳng, chuyển list, lấy độ dài.

from itertools import chain
listA = [[2,9,6], [5,'a'], [0], [12,'c',9,3]]
res = len(list(chain(*listA)))
print("Tổng số phần tử:", res)  # 10