python——列表生成式
阿新 • • 發佈:2020-12-24
列表生成式
1. 列表生成式(list comprehensions),是python內建的可以建立 list 的生成式
如:
生成一個list,【1,2,3,4,5,6,7,8,9,10】,可以用range()
>>> list(range(1,11))
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
若要生成【1*1,2*2,3*3,,,,10*10】
>>> for x in range(1,11): L.append(x*x) >>> L [1, 4, 9, 16, 25, 36, 49, 64, 81, 100]
但是迴圈很繁瑣,可以用列表生成式直接生成一個list:
>>> [x*x for x in range(1,11)]
[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]
2. 還可以使用兩層迴圈,生成全排列
#“+”代表連線符,將兩個字元連線起來
>>> [x+y for x in 'abc' for y in '123'] ['a1', 'a2', 'a3', 'b1', 'b2', 'b3', 'c1', 'c2', 'c3']
3. 用兩個變數生成一個list
>>> d = {'x': 'A', 'y': 'B', 'z': 'C' } >>> [k + '=' + v for k, v in d.items()] ['x=A', 'y=B', 'z=C']
4. 在使用列表生成式的時候 if 。。。else。。。的用法:
在for前面的是一個 if。。。else。。。表示式,
在for後面的 if 是過濾條件,不能帶 else
如:生成【1,2,3,4,5,6,7,8,9,10】中的偶數部分
>>> [x for x in range(1,11) if x%2==0]
[2, 4, 6, 8, 10]
#若將if寫在for前面,則會出錯
>>> [x if x%2==0 for x in range(1,11)]
SyntaxError: invalid syntax
在for前面加上if。。。else。。。的效果:
>>> [x if x%2==0 else 0 for x in range(1,11)]
[0, 2, 0, 4, 0, 6, 0, 8, 0, 10]
5. 將list中的大寫變成小寫,可以用lower()方法
用 isinstance()方法判斷是不是字串或者其他資料型別
如,有一個數據元素型別不同的list,選出其中字串型別的元素並將其大寫的內容改為小寫,建立一個新的list
>>> L=['A','B','c','d',123]
>>> [x.lower() for x in L if isinstance(x,str)]
['a', 'b', 'c', 'd']