大列表生成优化

以下内容似乎可以提高35%的速度:

import re

r = re.compile(r"(\w+)\[(\d+)(?::(\d+))?\]")

def expand_seq(seq):
    result = []
    for item in seq:
        m = r.match(item)
        name, start, end = m.group(1), int(m.group(2)), m.group(3)
        rng = xrange(start, int(end)) if end else (start,)
        t = name + "["
        result.extend(t + str(i) + "]" for i in rng)
    return result

使用此代码:

• 我们编译一个正则表达式用于函数。
• 我们直接连接字符串。

我不确定你会得到 戏剧性的 加速,因为无法从根本上改进算法。通过这样做,我确实比你的速度提高了20%:

def expand_seq(seq):
    expanded = []
    for s in seq:
        name, indices = s[0:-1].split("[")
        if ":" in indices:
            index1, index2 = [int(i) for i in indices.split(":")]
        else:
            index1 = int(indices)
            index2 = index1
        for n in range(index1, index2 + 1):
            expanded.append("{}[{}]".format(name, n))
    return expanded

我认为提速主要是因为没有重复一些操作(比如 int 和 split )你必须这样做才能将你的解决方案保持在一行。

正如已经建议的那样,如果使用生成器,您可以立即开始使用结果。这样地:

def expand_seq(seq):
    for s in seq:
        name, indices = s[0:-1].split("[")
        if ":" in indices:
            index1, index2 = [int(i) for i in indices.split(":")]
        else:
            index1 = int(indices)
            index2 = index1
        for n in range(index1, index2 + 1):
            yield "{}[{}]".format(name, n)