Python 高級特性（3）- 列表生成式

range() 函數

日常工作中，range() 應該非常熟悉了，它可以生成一個迭代對象，然后可以使用 list() 將它轉成一個 list

# 判斷是不是迭代對象
print(isinstance(range(0, 10), Iterable))

# 生成列表
lists = list(range(0, 10))
print(lists)

# 輸出結果
True
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

 

range 詳解

https://www.cnblogs.com/poloyy/p/15086994.html

 

具體更多的栗子

for i in range(5):
    print(i)

for i in range(10, 15):
    print(i)

for i in range(5, 25, 5):
    print(i)

 

運行結果

0
1
2
3
4

10
11
12
13
14

5
10
15
20

 

前置知識

for 循環：https://www.cnblogs.com/poloyy/p/15087053.html

if 語句： https://www.cnblogs.com/poloyy/p/15087130.html

 

列表生成式前置知識

• 但如果要生成 [1x1, 2x2, 3x3, ..., 10x10] 怎么做？
• 當你不會列表生成式前就是用循環

res = []
for i in range(11):
    res.append(i * i)
print(res)

# 輸出結果
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

但很明顯繁瑣了很多

 

列表生成式

針對上面的問題，列表生成式一句代碼即可完成

res = [i * i for i in range(11)]
print(res)

# 輸出結果
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

 

列表生成式：帶 if

# 篩選偶數
res = [i * i for i in range(11) if i % 2 == 0]
print(res)

# 輸出結果
[0, 4, 16, 36, 64, 100]

 

等價寫法

res = []
for i in range(11):
    if i % 2 == 0:
        res.append(i * i)

 

列表生成式：雙重循環

res = [i + j for i in range(5) for j in range(6, 11)]
print(res)

# 輸出結果
[6, 7, 8, 9, 10, 7, 8, 9, 10, 11, 8, 9, 10, 11, 12, 9, 10, 11, 12, 13, 10, 11, 12, 13, 14]

 

等價寫法

res = []
for i in range(5):
    for j in range(6, 11):
        res.append(i + j)

 

列表生成式：多個變量

d = {'x': 'A', 'y': 'B', 'z': 'C'}
res = [(k, v) for k, v in d.items()]
print(res)

# 輸出結果
[('x', 'A'), ('y', 'B'), ('z', 'C')]

 

等價寫法

d = {'x': 'A', 'y': 'B', 'z': 'C'}
res = []
for k, v in d.items():
    res.append((k, v))

 

列表生成式：包含函數

L = ['Hello', 'World', 'IBM', 'Apple']
res = [s.lower() for s in L]
print(res)

# 輸出結果
['hello', 'world', 'ibm', 'apple']

 

列表生成式：包含 if...else

錯誤寫法一

res = [i * i for i in range(11) if i % 2 == 0 else 0]
print(res)

# 報錯信息
res = [i * i for i in range(11) if i % 2 == 0 else 0]
　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　^
SyntaxError: invalid syntax

• pycharm 也會直接提示語法錯誤
• 因為跟在 for 后面的 if 是一個篩選條件，不能帶 else

 

錯誤寫法二

res = [i if i % 2 == 0  for i in range(11)]
print(res)

# 報錯信息
res = [i if i % 2 == 0  for i in range(11)]
                                  ^
SyntaxError: invalid syntax

for 前面是一個表達式，它需要有個表達式計算結果，只有 if 是不完整的，需要有 else

 

正確寫法

res = [i if i % 2 == 0 else "奇數" for i in range(11)]
print(res)

# 輸出結果
[0, '奇數', 2, '奇數', 4, '奇數', 6, '奇數', 8, '奇數', 10]

將 if ... else 放在前面寫