Python - 列表解析式

列表解析——用來動態地建立列表

[expr for iter_var in iterable if cond_expr]

例子一：

 

map(lambda x: x**2, range(6))

[0, 1, 4, 9, 16, 25]

[x**2 for x in range(6)]

[0, 1, 4, 9, 16, 25]

列表解析式能夠取代內建的map()函數以及lambda，並且++效率更高++。

例子二：

seq = [11, 10, 9, 8, 7, 6]

filter(lambda x: x % 2, seq)

[11, 9, 7]

[x for x in seq if x % 2]

[11, 9, 7]

例子三：

找出文本中最長的詞

# 第一個最長的詞
longest = ''
for word in text:
    if len(word) > len(longest):
        longest = word

#全部最長的詞
maxlen = max(len(word) for word in text) # 生成器表達式

[word for word in text if len(word) == maxlen] #鏈表推導式

不足： 列表解析得一個不足就是必要生成全部數據，用於建立整個列表。這可能對有大量數據得迭代其有負面效應。

++解決方法++：生成器表達式，經過結合列表解析和生成器解決了這個問題。

嵌套列表推導式

例一

all_data = [[1,3,5,7,9],
            [2,4,6,8,10]]

result = []
for nums in all_data:
    temp_num = [num for num in nums if num >= 5]
    result.extend(temp_num)

嵌套列表推導式

result = [num for nums in all_data for num in nums if num >= 5]

推導式中for的部分是按嵌套順序排列的。

例二

some_tuples = [(1,2,3), (4,5,6), (7,8,9)]

flattened = [x for tup in some_tuples for x in tup]

flattened
[1,2,3,4,5,6,7,8,9]

 列表解析配合if else

列表解析總共有兩種形式：

1. [i for i in range(k) if condition]：此時if起條件判斷做用，知足條件的，將被返回成爲最終生成的列表的一員。

2. [i if condition else exp for exp]：此時if...else被用來賦值，知足條件的i以及else被用來生成最終的列表。

以上狀況對多個for仍然成立。

print([i for i in range(10) if i%2 == 0])  
print([i if i == 0 else 100 for i in range(10)])  
  
[0, 2, 4, 6, 8]  
[0, 100, 100, 100, 100, 100, 100, 100, 100, 100]  

 

 

 

參考來源：http://blog.csdn.net/shingle_/article/details/55050701

　　　　　http://blog.csdn.net/zk_j1994/article/details/72809260