我想從一個列表中刪除所有零，並將它們全部添加到另一列表中，但不是全部刪除

Question

我想從一個列表中刪除所有零，並將它們附加到另一列表中，但是當一個以上的零彼此接連出現時，其中一個不會刪除。

zero=[]
l=[2,3,6,0,0,5,6,0,7]
for i in l:
    if i==0:
        l.remove(i)
        zero.append(i)
print(l)
print(zero)


l=[2, 3, 6, 5, 6, 0, 7]
zer0=[0, 0]

在此輸出中，零之一不刪除

Answer 1

迭代對象時不要修改列表。 相反，您可以使用副本。 l[:]

例如：

zero=[]
l=[2,3,6,0,0,5,6,0,7]
for i in l[:]:
    if i==0:
        l.remove(i)
        zero.append(i)
print(l)
print(zero)

Answer 2

您可以從numpy模塊中使用non_zero 。 它旨在查找所有不同於0的值。

如果要得到所有零，則可以使用np.where找到它們

l = [2, 3, 6, 0, 0, 5, 6, 0, 7]

import numpy as np
np_l = np.array(l)

non_zero = np_l[np.nonzero(np_l)]
print(non_zero)
# [2 3 6 5 6 7]

zero = np_l[np.where(np_l == 0)]
print(zero)
# [0 0 0]

Answer 3

您正在遍歷正在修改的列表。 如果從該列表中刪除當前元素，它將縮小並且next(l)將跳過一個元素。 為了避免這種情況，請始終遍歷列表的副本。 例如list(l)或l[:] 。

zero=[]
l=[2,3,6,0,0,5,6,0,7]
for i in list(l):
    if i==0:
        l.remove(i)
        zero.append(i)
print(l)
print(zero)

Answer 4

使用itertools.groupby ：

from itertools import groupby

l=[2,3,6,0,0,5,6,0,7]

non_zeros = []
zeros = []
for v, g in groupby(l, lambda k: k==0):
    if v:
        zeros.extend(g)
    else:
        non_zeros.extend(g)

print(non_zeros)
print(zeros)

印刷品：

[2, 3, 6, 5, 6, 7]
[0, 0, 0]