python - 将所有零移动到 Python 列表的开头

Question

我有如下列表:

a = [4, 5, 0, 0, 6, 7, 0, 1, 0, 5]

我想将所有零推到该列表的开头。结果必须如下所示。

a = [0, 0, 0, 0, 4, 5, 6, 7, 1, 5]

如何在 Python 2 中实现？

Answer 1

您可以对列表进行排序:

a.sort(key=lambda v: v != 0)

key 函数告诉 Python 根据值是否为 0 对值进行排序。 False 排在 True 之前，然后根据值的原始相对位置对值进行排序。

对于 0，返回 False，首先对所有这些值进行排序。对于其余的 True 返回，保留排序将它们放在最后但保持它们的相对位置不变。

演示:

>>> a = [4, 5, 0, 0, 6, 7, 0, 1, 0, 5]
>>> a.sort(key=lambda v: v != 0)
>>> a
[0, 0, 0, 0, 4, 5, 6, 7, 1, 5]