python - 将所有非零值放在数组的右边

Question

我有一个包含整数的 n 个元素的固定大小数组。我想将所有非零值放在数组的右侧。

例子:

[1, 0, 0, 4] => [0, 0, 1, 4]
[2, 0, 4, 3] => [0, 2, 4, 3]
[1, 0, 0] => [0, 0, 1]

执行此操作的简单算法是什么？

这是我目前拥有的，但它只会将索引处的元素向右移动

def slide(arr):

    i = len(arr) - 2
    while i >= 0:

        if arr[i] != 0 and arr[i + 1] == 0:
            arr[i + 1] = arr[i]
            arr[i] = 0

        i -= 1

    print(arr)

Answer 1

我会计算零，删除它们并将它们包含在开头:

zeros = [0] * arr.count(0)
while 0 in arr:
    arr.remove(0)
arr = zeros + arr