【Python】考虑用生成器改写直接返回列表的函数

使用生成器的好处是显而易见的，可以使代码更加清晰，同时减小内存的消耗，当函数需要返回列表，把函数改写为生成器是相对容易的。

下面这两个函数返回字符串中每个单词的索引：

def index_words1(text):
    result = []
    if text:
        result.append(0)
    for index, letter in enumerate(text):
        if letter == ' ':
            result.append(index+1)
    return result


def index_words2(text):
    if text:
        yield 0
    for index, letter in enumerate(text):
        if letter == ' ':
            yield index+1


if __name__ == '__main__':
    hhh = 'Every dog has its day!'
    result = index_words1(hhh)
    print('result1', result)
    result = list(index_words2(hhh))
    print('result2:', result)

 参考资料：Effective Python