python基础知识——列表表达式

列表表达式

1. 基本用法

如下图所示我们想输入参数的每个字母，输出到一个列表里

def output_letter(letter):

    l = []

    for item in letter:

        l.append(item)

    return l

if __name__ == "__main__":

    print(output_letter('levi'))

#此方法的输出为：

['l', 'e', 'v', 'i']

这样写很冗余，我们可以通过列表表达式来简化代码

def output_letter(letter):

   return [l for l in letter]



if __name__ == "__main__":

    print(output_letter('levi'))



#此方法的输出为：

['l', 'e', 'v', 'i']

2. 增加条件判断

a=[i for i in range(10) if i%2==0]
print(a)
b=[i if i%2==0 else 'll' for i in range(10)]
print(b)
#此方法输出的是：
[0, 2, 4, 6, 8]
[0, 'll', 2, 'll', 4, 'll', 6, 'll', 8, 'll']