美文网首页
列表推导式

列表推导式

作者: 脏脏包盛 | 来源:发表于2017-12-17 17:53 被阅读0次

    循环生成列表

    values = [10, 21, 4, 7, 12]
    squares = []
    for x in values:
        squares.append(x**2)
    print squares
    

    [100, 441, 16, 49, 144]
    列表推导式可以使用更简单的方法来创建这个列表:

    values = [10, 21, 4, 7, 12]
    squares = [x**2 for x in values]
    print(squares)
    

    [100, 441, 16, 49, 144]
    还可以在列表推导式中加入条件进行筛选。

    例如在上面的例子中,假如只想保留列表中不大于10的数的平方:

    values = [10, 21, 4, 7, 12]
    squares = [x**2 for x in values if x <= 10]
    print(squares)
    

    [100, 16, 49]
    也可以使用推导式生成集合和字典:

    square_set = {x**2 for x in values if x <= 10}
    print(square_set)
    square_dict = {x: x**2 for x in values if x <= 10}
    print(square_dict)
    

    set([16, 49, 100])
    {10: 100, 4: 16, 7: 49}

    再如,计算上面例子中生成的列表中所有元素的和:

    total = sum([x**2 for x in values if x <= 10])
    print(total)
    

    165
    但是,Python会生成这个列表,然后在将它放到垃圾回收机制中(因为没有变量指向它),这毫无疑问是种浪费。
    为了解决这种问题,Python使用产生式表达式来解决这个问题:

    total = sum(x**2 for x in values if x <= 10)
    print(total)
    

    165
    与上面相比,只是去掉了括号,但这里并不会一次性的生成这个列表。
    比较一下两者的用时:

    x = range(1000000)
    
    %timeit total = sum([i**2 for i in x])
    

    1 loops, best of 3: 3.86 s per loop

    %timeit total = sum(i**2 for i in x)
    

    1 loops, best of 3: 2.58 s per loop

    相关文章

      网友评论

          本文标题:列表推导式

          本文链接:https://www.haomeiwen.com/subject/skfuwxtx.html