Python：我怎样才能摆脱每个子列表的第二个元素？

Question

我有一个子列表列表，例如：

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

如何摆脱每个子列表的第二个元素？ （即4）

[501,501,501,501]

我应该迭代列表还是有更快的方法？ 感谢

Answer 1

您可以使用列表推导来获取每个子列表的第一个元素：

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

Answer 2

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

Answer 3

使用map的较少pythonic功能版本：

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

较少pythonic，因为它不使用列表推导。请参阅here。