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

Lấy phần tử đầu của mỗi danh sách con trong Python

Cho danh sách lồng nhau, lấy phần tử đầu tiên (chỉ số 0) của mỗi danh sách con.

Cách 1: Vòng lặp for

Alist = [['Mon', 1], ['Tue', 'Wed', "Fri"], [12,3,7]]
print("Given List:\n", Alist)
print("First Items from sublists:\n")
for item in Alist:
    print(item[0])

Kết quả: Mon Tue 12

Cách 2: Dùng zip và *

Alist = [['Mon', 1], ['Tue', 'Wed', "Fri"], [12,3,7]]
print("Given List:\n", Alist)
print("\n First Items from sublists:\n")
print(list(list(zip(*Alist))[0]))

Kết quả: ['Mon', 'Tue', 12]

Cách 3: Dùng itemgetter

from operator import itemgetter
Alist = [['Mon', 1], ['Tue', 'Wed', "Fri"], [12,3,7]]
print("Given List:\n", Alist)
print("\n First Items from sublists:\n")
print(list(map(itemgetter(0), Alist)))

Kết quả: ['Mon', 'Tue', 12]