盤點Python列表生成式的三種方法
一、前言
列表生成式即List Comprehensions，是Python內置的非常簡單卻強大的可以用來創建list的生成式。
二、案例分析
三種方法
要生成list [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]可以用list(range(1, 11))。
- print(list(range(1, 11)))
如果要生成[1x1, 2x2, 3x3, …, 10x10]怎么做?
1. 方法一是循環：
- L = []
- for x in range(1, 11):
- L.append(x * x)
- print(L)
但是循環太繁瑣，而列表生成式則可以用一行語句代替循環生成上面的list：
- print([x * x for x in range(1, 11)])
寫列表生成式時，把要生成的元素x * x放到前面，后面跟for循環，就可以把list創建出來，十分有用，多寫幾次，很快就可以熟悉這種語法。
for循環后面還可以加上if判斷，這樣就可以篩選出僅偶數的平方：
- for x in range(1, 11):
- L.append(x * x)
- print([x * x for x in range(1, 11) if x % 2 == 0])
2. 使用兩層循環，可以生成全排列
- L = []
- for x in range(1, 11):
- L.append(x * x)
- print( [m + n for m in 'ABC' for n in 'XYZ'])
三層和三層以上的循環就很少用到了。
3. 運用列表生成式，可以寫出非常簡潔的代碼。
例如，列出當前目錄下的所有文件和目錄名，可以通過一行代碼實現：
- import os # 導入os模塊，模塊的概念后面講到
- print([d for d in os.listdir('.')]) # os.listdir可以列出文件和目錄
for循環其實可以同時使用兩個甚至多個變量，s比如dict的items()可以同時迭代key和value：
- d = {'x': 'A', 'y': 'B', 'z': 'C'}
- for k, v in d.items():
- print(k, '=', v)
因此，列表生成式也可以使用兩個變量來生成list：
- d = {'x': 'A', 'y': 'B', 'z': 'C' }
- print([k + '=' + v for k, v in d.items()]
最后把一個list中所有的字符串變成小寫：
- L = ['Hello', 'World', 'IBM', 'Apple']
- print([s.lower() for s in L])
如果list中既包含字符串，又包含整數，由于非字符串類型沒有lower()方法，所以列表生成式會報錯：
- L = ['Hello', 'World', 18, 'Apple', None]
- print([s.lower() for s in L])
使用內建的isinstance函數可以判斷一個變量是不是字符串：
- x = 'abc'
- y = 123
- print(isinstance(x, str))
- print(isinstance(y, str))
三、練習和思考
請修改列表生成式，通過添加if語句保證列表生成式能正確地執行。
- # -*- coding: utf-8 -*-
- L1 = ['Hello', 'World', 18, 'Apple', None]
- L2=???
- # 期待輸出: ['hello', 'world', 'apple']
- print(L2)
練習參考代碼 :
- L2 =[s.lower() for s in L1 if isinstance(s, str) ]
運行結果：
注:
運用列表生成式，可以快速生成list，可以通過一個list推導出另一個list，而代碼卻十分簡潔。
四、總結
本文基于Python基礎，介紹了列表生成式，以案例為主，三種方法進行講解。
分析不同方法的實現一樣的效果。實際案例中需要注意的點，遇到的難點，提供有效的解決方法。最后通過練習和思考，對列表生成式加深理解。
歡迎大家積極嘗試，有時候看到別人實現起來很簡單，但是到自己動手實現的時候，總會有各種各樣的問題，切勿眼高手低，勤動手，才可以理解的更加深刻。
代碼很簡單，希望對你學習有幫助。
