函数-函数进阶-列表生成式

>>> a = list(range(10))
>>> a
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
>>> a = [i+1 if i<10 else i*i for i in a]
>>> a
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
>>>