Python列表生成式之條件過濾

列表生成式的 for 循環後面還可以加上 if 判斷。例如:

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

如果我們只想要偶數的平方,不改動 range()的情況下,可以加上 if 來篩選:

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

有了 if 條件,只有 if 判斷爲 True 的時候,才把循環的當前元素添加到列表中。

 

練習:請編寫一個函數,它接受一個 list,然後把list中的所有字符串變成大寫後返回,非字符串元素將被忽略。

提示:

1. isinstance(x, str) 可以判斷變量 x 是否是字符串;

2. 字符串的 upper() 方法可以返回大寫的字母。

代碼參考:
L = [x * x for x in range(1, 11)]
print(L)

#如果我們只想要奇數的平方,不改動 range()的情況下,可以加上 if 來篩選
L1 = [x * x for x in range(1, 11) if x % 2 == 1]
print(L1)
'''有了 if 條件,只有 if 判斷爲 True 的時候,才把循環的當前元素添加到列表中'''


#編寫一個函數,它接受一個 list,然後把list中的所有字符串變成大寫後返回,非字符串元素將被忽略
def toUpper(L):
    return [x.upper() for x in L if isinstance(x, str)]
L1 = ['Hello', 0, 'RaoJuan', 317]
print(toUpper(L1))

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章