1. 程式人生 > >Python - 列表解析式

Python - 列表解析式

map som 最長 順序 flat pan lte data 列表

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

[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

Python - 列表解析式