Python 删除列表中序列的内部项

Question

foo = [0,0,0,0,1,1,1,1,1,1,0,0,0,1,1,1,0,0,1,1]

bar = [x if x==0 else 'o' for x in foo]

酒吧：

[0, 0, 0, 0,'o', 'o', 'o', 'o', 'o', 'o', 0, 0, 0, 'o', 'o', 'o', 0, 0, 'o', 'o']

我想此时删除内部的“o”，以便结果如下所示：

[0, 0, 0, 0, 'o','o', 0, 0, 0, 'o','o', 0, 0, 'o', 'o']

如果可能，我想在列表理解本身中执行此操作，并且我想避免任何转换为字符串的操作（因为我的实际任务涉及字典而不是 1 和 0）。有任何想法吗？

Answer 1

如果你真的想用一个理解列表来做：bar=[x if x==0 else 'o' for i,x in enumerate(foo) if (i==0 or i==len(foo)-1) or x==0 or foo[i-1]==0 or foo[i+1]==0]应该适用于您的示例。

Answer 2

你可以做：>>> foo = [0,0,0,0,1,1,1,1,1,1,0,0,0,1,1,1,0,0,1,1]>>> from itertools import groupby>>> [ext for c, grp in groupby(foo) for ext in (grp if c==0 else ['o']*min(2,len(list(grp))))][0, 0, 0, 0, 'o', 'o', 0, 0, 0, 'o', 'o', 0, 0, 'o', 'o']