根据编写函数,检查传入列表的长度,如果大于2那仅仅保留前两个长度的内容,并将新内容返回
普通写法
def remain_one(mlist):
if len(mlist) > 2:
return mlist[:2]
return mlist
print(remain_one(mlist=[3,4,6,9]))
三元表达式
def remain_two(mlist):
return mlist[:2] if len(mlist) > 2 else mlist
print(remain_two(mlist=[1,3,4,8]))
源码中出现的情况比较多(简写代码),和if else一样,只有一个表达式会被执行。因此三元表达式中的if和else可以包含大量的计算,但只有True的分支会被执行