3

我有一个代表数据结构遍历的字符串列表。我想将链接列表遍历折叠成更紧凑的表示。为此，我想计算相邻next和prev链接的数量并将它们折叠成一个整数。

以下是我想做的转换示例：

['modules']                                   -->  ['modules']
['modules', 'next']                           -->  ['modules', 1]
['modules', 'prev']                           -->  ['modules', -1]
['modules', 'next', 'next', 'next', 'txt']    -->  ['modules', 3, 'txt']
['modules', 'next', 'prev', 'next', 'txt']    -->  ['modules', 1, 'txt']
['super_blocks', 'next', 's_inodes', 'next']  -->  ['super_blocks', 1, 's_inodes', 1]

每个next链接计为+1，每个链接prev为-1。相邻next的 s 和prevs 相互抵消。他们可以按任何顺序出现。

我有一个可行的解决方案，但我正在努力寻找一个令人满意的优雅和 Pythonic 的解决方案。

4

4 回答 4

6

您可以使用生成器：

def links(seq):
    it = iter(seq)
    while True:
        el = next(it)
        cnt = 0
        try:
            while el in ['prev', 'next']:
                cnt += (1 if el == 'next' else -1)
                el = next(it)
        finally:
            if cnt != 0:
                yield cnt
        yield el

print list(links(['modules', 'next', 'prev', 'next', 'txt']))

值得注意的是，包含相同数量next和的序列prev会被完全删除。如果这是您想要的，则可以很容易地更改代码以生成 a 0（我认为对此的要求还不清楚）。

于 2013-03-21T21:49:40.050 回答
1

这是我想到的最直接的方法。直截了当是理解、调试和未来维护的宝贵品质。

def process(l):
    result = []
    count = 0
    keepCount = False
    for s in l:
        if s == "next":
            count += 1
            keepCount = True
        elif s == "prev":
            count -= 1
            keepCount = True
        else:
            if keepCount:
                result.append(count)
                count = 0
                keepCount = False
            result.append(s)
        # end if
    # end for
    if keepCount:
        result.append(count)

    return result
# end process()

不过，我确实更喜欢 NPE 使用生成器。（我的可以通过将'result.append（）'更改为'yield'来轻松转换）他的（原始）答案与我的几乎相同，但如果下一个/上一个标记相邻，我会包括0计数数量相等。

于 2013-03-21T21:54:52.383 回答
1

一点点怎么样reduce()？

def collapse(lst):
    was_link = [False] # No nonlocal in Python 2.x :(
    def step(lst, item):
        val = { 'prev': -1, 'next': 1 }.get(item)

        if was_link[0] and val:
            lst[-1] += val
        else:
            lst.append(val or item)
        was_link[0] = bool(val)

        return lst

    return reduce(step, [[]] + lst)
于 2013-03-21T22:12:00.900 回答
1

怎么样：

def convert(ls):
    last = None
    for x in ls:
        if x == 'prev': x = -1
        if x == 'next': x = +1
        if isinstance(x, int) and isinstance(last, int):
            x += last
        elif last:  # last is not None if you want zeroes
            yield last
        last = x
    yield last
于 2013-03-21T22:15:12.460 回答