根据值和规则拆分整数列表的更好方法

时间:2019-04-23 05:36:59

标签: python list loops integer sublist

目标是根据以下规则,根据每个元素的邻居拆分整数列表:

  • (当前/焦点值小于或等于前一个)和(当前值等于下一个),即prev_value >= focal_value == next_value

  • (当前值小于前一个值)和(当前值小于下一个值),即prev_value > focal_value < next_value

举例说明,给定x产生y

x = (1, 4, 2, 1, 4, 2, 4, 1, 4, 1, 4, 4, 3)
y = [[1, 4, 2], [1, 4], [2, 4], [1, 4], [1, 4, 4, 3]]
assert func(x) == y

我尝试过:

def per_window(sequence, n=1):
    """
    From http://stackoverflow.com/q/42220614/610569
        >>> list(per_window([1,2,3,4], n=2))
        [(1, 2), (2, 3), (3, 4)]
        >>> list(per_window([1,2,3,4], n=3))
        [(1, 2, 3), (2, 3, 4)]
    """
    start, stop = 0, n
    seq = list(sequence)
    while stop <= len(seq):
        yield tuple(seq[start:stop])
        start += 1
        stop += 1


def func(x):
    result = []
    sub_result = [x[0]]
    for prev_value, focal_value, next_value in per_window(x, 3):
        # These if and elif cases trigger syllable break.
        if prev_value >= focal_value == next_value:
            sub_result.append(focal_value)
            result.append(sub_result)
            sub_result = []
        elif prev_value > focal_value < next_value:
            result.append(sub_result)
            sub_result = []
            sub_result.append(focal_value)
        else: # no  break
            sub_result.append(focal_value)
    sub_result.append(next_value)
    result.append(sub_result)
    return result


x = (1, 4, 2, 1, 4, 2, 4, 1, 4, 1, 4, 4, 3)
y = [[1, 4, 2], [1, 4], [2, 4], [1, 4], [1, 4, 4, 3]]

assert func(x) == y

但是我的问题是:

  • 如果我们仔细看一下if和elif的“ cases”,它看起来就像第一个if永远不会被捕获,因为第二个if将首先到达。那正确吗?首先发生if个案例的例子有哪些?

  • 是否有更好的方法来实现基于规则拆分子列表的相同目标?

  • 最后两个附加项需要存在于per_window的循环之外,这些扼杀者叫什么?有没有不这样做的循环方法?

1 个答案:

答案 0 :(得分:1)

  1. 在您的示例中,第一个if永远不会触发。但是这个示例数据将是这样:

    x = (5 ,4, 4, 2)
    y = [[5, 4], [4, 2]]
    
  2. 更好是主观的。但是,只要不移动窗口,只需移动值就可以达到相同的目标

    def func2(x):
        x = iter(x)
        try:
            prev_value = next(x)
            focal_value = next(x)
        except StopIteration:
            return [list(x)]
        sub_result = [prev_value]
        result = [sub_result]
        for next_value in x:
            if prev_value >= focal_value == next_value:
                sub_result.append(focal_value)
                sub_result = []
                result.append(sub_result)
            elif prev_value > focal_value < next_value:
                sub_result = [focal_value]
                result.append(sub_result)
            else:
                sub_result.append(focal_value)
            prev_value, focal_value = focal_value, next_value
        sub_result.append(focal_value)
        return result
    

    timeit说,速度快两倍以上

  3. 在循环中保留最后一个值后,便会在循环后为其添加特殊处理。但是我的代码表明可以将sub_result列表添加到循环中。