在保持结构的同时重置嵌套列表

时间:2014-04-02 10:15:56

标签: python algorithm list python-3.x nested-lists

是否可以在保持“结构”的同时“清除”python中的(嵌套)列表。我的意思是,如果输入它是

[1,3,4,[5,2,[4,3],3],6,[7,8],9]

输出为

[0,0,0,[0,0,[0,0],0],0,[0,0],0]

保持列表的实际结构并重置值。

上下文:实际上没有任何问题,我想到了这个问题,我想知道是否可以做到。

4 个答案:

答案 0 :(得分:3)

这是一个利用duck typing原则的实现:

In [5]: def clear(seq):
   ...:     for i, item in enumerate(seq):
   ...:         try:
   ...:             clear(item)
   ...:         except TypeError:
   ...:             try:
   ...:                 seq[i] = 0
   ...:             except IndexError, KeyError:
   ...:                 pass
   ...:             

In [6]: s = [1,3,4,[5,2,[4,3],3],6,[7,8],9]

In [9]: clear(s)

In [10]: s
Out[10]: [0, 0, 0, [0, 0, [0, 0], 0], 0, [0, 0], 0]

这样做的好处是,您不受限于intlist s的结构/值。这个功能将为零'任何支持索引的可变结构。

答案 1 :(得分:2)

>>> l = [1,3,4,[5,2,[4,3],3],6,[7,8],9]
>>> def reset(lst):
...     return [reset(e) if isinstance(e, list) else 0 for e in lst]
...
>>> l = reset(l)
>>> l
[0, 0, 0, [0, 0, [0, 0], 0], 0, [0, 0], 0]

答案 2 :(得分:1)

我使用递归工作,虽然它有点原始。

注意:仅限整数和列表,但可以是任意随机嵌套。适用于所有人

def set_zero(lst):
    for i,element in enumerate(lst):
        if type(element) != list: lst[i]=0
        else: set_zero(element)

>>> a = [1,3,4,[5,2,[4,3],3],6,[7,8],9]
>>> set_zero(a)
>>> print(a)
[0,0,0,[0,0,[0,0],0],0,[0,0],0]

顺便问一下好问题!

答案 3 :(得分:1)

我能够找到另一种方法,基本上是作弊但它有效 此外,它是单行 !!

def set_zero(lst):
    return eval(''.join(str(0) if x.isdigit() else str(x) for x in str(lst)))

>>> a = [1,3,4,[5,2,[4,3],3],6,[7,8],9]
>>> a = set_zero(a)
>>> print(a)
[0,0,0,[0,0,[0,0],0],0,[0,0],0]

缺点: 它比我之前的建议慢了3/4。

Method 1: ~0.000035 s
Method 2: ~0.000120 s

我发布它作为一个不同的答案,因为它是一个完全不同的概念。如果你愿意,我可以将它合并回我的第一个答案。