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]