如何将python列表或numpy数组中的所有非零元素移到一侧？

Question

我将对列表或numpy数组执行以下操作：

[0, 0, 0, 1, 0, 0, 4, 2, 0, 7, 0, 0, 0]

将所有非零值移到右侧：

[0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 4, 2, 7]

我如何有效地做到这一点？

谢谢

============

抱歉，我不清楚，我需要保留非零元素的顺序。

Answer 1

您可以按其布尔值对列表进行排序。 所有伪造的值（数字仅为零）将被推到列表的最前面。 Python的内置排序看起来很稳定，因此其他值将保持其相对位置。

例：

>>> a = [0, 0, 0, 1, 0, 0, 5, 2, 0, 7, 0, 0, 0]
>>> sorted(a, key=bool)
[0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 5, 2, 7]

Answer 2

使用NumPy：

>>> a = np.array([0, 0, 0, 1, 0, 0, 4, 2, 0, 7, 0, 0, 0])
>>> np.concatenate((a[a==0], a[a!=0]))
array([0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 4, 2, 7])

您也可以通过使用简单的for循环在Python的O(N)时间执行此操作。 但是会占用一些额外的内存，我们可以使用a.sort(key=bool)的解决方案中避免：

>>> from collections import deque
#Using a deque
>>> def solve_deque(lst):
    d = deque()
    append_l = d.appendleft
    append_r = d.append
    for x in lst:
        if x:
            append_r(x)
        else:
            append_l(x)
    return list(d) #Convert to list if you want O(1) indexing.
...
#Using simple list
>>> def solve_list(lst):                           
    left = []                                    
    right = []
    left_a = left.append
    right_a = right.append
    for x in lst:
        if x:
            right_a(x)
        else:
            left_a(x)
    left.extend(right)
    return left

>>> solve_list([0, 0, 0, 1, 0, 0, 4, 2, 0, 7, 0, 0, 0])
[0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 4, 2, 7]
>>> solve_deque([0, 0, 0, 1, 0, 0, 4, 2, 0, 7, 0, 0, 0])
[0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 4, 2, 7]