美文网首页
Python高级特性——列表生成式

Python高级特性——列表生成式

作者: 东伶 | 来源:发表于2021-03-23 08:13 被阅读0次

    列表生成式即List Comprehensions,是Python内置的非常简单却强大的可以用来创建list的生成式。
    举个例子,要生成list [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]可以用list(range(1, 11)):

    >>> list(range(1, 11))
    [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
    

    但如果要生成[1x1, 2x2, 3x3, ..., 10x10]怎么做?

    >>> [x * x for x in range(1, 11)]
    [1, 4, 9, 16, 25, 36, 49, 64, 81, 100]
    

    for循环后面还可以加上if判断,这样我们就可以筛选出仅偶数的平方:

    >>> [x * x for x in range(1, 11) if x % 2 == 0]
    [4, 16, 36, 64, 100]
    

    还可以使用两层循环,可以生成全排列:

    >>> [m + n for m in 'ABC' for n in 'XYZ']
    ['AX', 'AY', 'AZ', 'BX', 'BY', 'BZ', 'CX', 'CY', 'CZ']
    if ... else
    

    使用列表生成式的时候,有些童鞋经常搞不清楚if...else的用法。
    跟在for后面的if是一个筛选条件,不能带else,如:

    >>> [x for x in range(1, 11) if x % 2 == 0]
    [2, 4, 6, 8, 10]
    

    for前面的部分是一个表达式,它必须根据x计算出一个结果。因此,必须加上else,如:

    >>> [x if x % 2 == 0 else -x for x in range(1, 11)]
    [-1, 2, -3, 4, -5, 6, -7, 8, -9, 10]

    相关文章

      网友评论

          本文标题:Python高级特性——列表生成式

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