使用列表推导从嵌套的Python数据结构中获取值列表?

时间:2013-08-17 07:14:55

标签: python data-structures

在使用耐心排序计算最长的增加子序列时,我创建了一个(值,反向链接)对的嵌套元组,然后我需要抓取它以按顺序获取所有值。 目前我使用一个小函数来做到这一点但是想知道是否有办法使用列表推导来做同样的事情。

该示例在我的原始程序中使用了一个namedtuple,应该说明我的问题:

>>> from collections import namedtuple
>>> P = namedtuple('P', 'val, back')
>>> q = P(val=15, back=P(val=11, back=P(val=9, back=P(val=6, back=P(val=2, back=P(val=0, back=None))))))
>>> q
P(val=15, back=P(val=11, back=P(val=9, back=P(val=6, back=P(val=2, back=P(val=0, back=None))))))
>>> #Request something like:
>>> [val for val, q in q]
Traceback (most recent call last):
  File "<pyshell#68>", line 1, in <module>
    [val for val, q in q]
  File "<pyshell#68>", line 1, in <listcomp>
    [val for val, q in q]
TypeError: 'int' object is not iterable
>>> q
P(val=15, back=P(val=11, back=P(val=9, back=P(val=6, back=P(val=2, back=P(val=0, back=None))))))
>>> # Wanted: [15, 11, 9, 6, 2, 0]
>>> 
>>> # Have to use:
>>> def _unwind(q):
    u = []
    while q:
        u.append(q.val)
        q = q.back
    return u

>>> q
P(val=15, back=P(val=11, back=P(val=9, back=P(val=6, back=P(val=2, back=P(val=0, back=None))))))
>>> _unwind(q)
[15, 11, 9, 6, 2, 0]
>>> 

我正在寻找一些设置P的方法,所以我可以编写类似的东西:

[val for val, q in q]

也许通过某种方式重写P .__ iter__和P .__ next__来优雅地重新定义P的iter协议?

1 个答案:

答案 0 :(得分:2)

请尝试以下操作:

>>> from collections import namedtuple
>>>
>>> class P(namedtuple('_P', 'val back')):
...     def __iter__(self):
...         while self:
...             yield self.val, self.back
...             self = self.back
...
>>> q = P(val=15, back=P(val=11, back=P(val=9, back=P(val=6, back=P(val=2, back=P(val=0, back=None))))))
>>> [val for val, back in q]
[15, 11, 9, 6, 2, 0]