在使用耐心排序计算最长的递增子序列时,我创建了一个(值,反向链接)对的嵌套元组,然后我需要对其进行爬网以按顺序获得所有值。目前,我使用一个小函数来执行此操作,但我想知道是否有一种方法可以使用列表推导来执行相同的操作。
该示例使用了我的原始程序中的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协议?
请尝试以下操作:
>>> 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]
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句