Python: как я могу избавиться от второго элемента каждого подсписок?

Question

У меня есть список подсписков, таких как:

[[501, 4], [501, 4], [501, 4], [501, 4]]

Как я могу избавиться второго элемента для каждого подсписника? (Т.е. 4)

[501, 501, 501, 501]

Должен ли я просматривать список или есть более быстрый способ? благодаря

Answer 1

a = [[501, 4], [501, 4], [501, 4], [501, 4]] 
b = [c[0] for c in a] 

Answer 2

Вы можете использовать список понимание, чтобы сделать первый элемент каждого подсписка:

xs = [[501, 4], [501, 4], [501, 4], [501, 4]] 
[x[0] for x in xs] 
# [501, 501, 501, 501] 

Answer 3

Менее вещий, функциональная версия с использованием map:

a = [[501, 4], [501, 4], [501, 4], [501, 4]] 
map(lambda x: x[0], a) 

Меньше вещий, поскольку он не использует списки. См. here.