列表生成式的 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))