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

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

Cho danh sách lồng nhau (chứa các danh sách con), lấy phần tử cuối 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("Last Items from sublists:\n")
for item in Alist:
    print(item[-1])

Kết quả: 1 Fri 7

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(*map(reversed, Alist)))[0]))

Kết quả: [1, 'Fri', 7]

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(-1), Alist)))

Kết quả: [1, 'Fri', 7]