Python 語言裡有許多(而且是越來越多)的高級特性,是 Python 發燒友們非常喜歡的。在這些人的眼裡,能夠寫出那些一般開發者看不懂的高級特性,就是高手,就是大神。
但你要知道,在團隊合作裡,炫技是大忌。
為什麼這麼說呢?我說下自己的看法:
越簡潔的代碼,越清晰的邏輯,就越不容易出錯;
在團隊合作中,你的代碼不只有你在維護,降低別人的閱讀/理解代碼邏輯的成本是一個良好的品德
簡單的代碼,只會用到最基本的語法糖,復雜的高級特性,會有更多的依賴(如語言的版本)
該篇是「炫技系列」的第三篇內容,在這個系列裡,我將總結盤點一下,我所見過的那些炫技操作。在這裡,如果你是 Python 發燒友,你可以學到一些寫出超酷的代碼書寫技巧。同時,看了這些內容,對你在閱讀別人的代碼時,也許會有些幫助。
>>> list01 = [1,2,3]
>>> list02 = [4,5,6]
>>> list03 = [7,8,9]
>>>
>>> list01 + list02 + list03
[1, 2, 3, 4, 5, 6, 7, 8, 9]
在前面的文章中也介紹過,使用 itertools.chain() 函數先可迭代對象(在這裡指的是列表)串聯起來,組成一個更大的可迭代對象。
最後你再利用 list 將其轉化為 列表。
>>> from itertools import chain
>>> list01 = [1,2,3]
>>> list02 = [4,5,6]
>>> list03 = [7,8,9]
>>>
>>> list(chain(list01, list02, list03))
[1, 2, 3, 4, 5, 6, 7, 8, 9]
與它相似的,使用 * 可以解包列表。 * 和 ** 常用在函數定義時,設置可變參數。
現在我將它單獨拿出來用在多個列表的合並。
示例如下:
>>> list01 = [1,2,3]
>>> list02 = [4,5,6]
>>>
>>> [*list01, *list02]
[1, 2, 3, 4, 5, 6]
>>> list01 = [1,2,3]
>>> list02 = [4,5,6]
>>>
>>> list01.extend(list02)
>>> list01
[1, 2, 3, 4, 5, 6]
那就是列表解析式,集合解析式和字典解析式,通常是 Python 發燒友的最愛,那麼今天的主題:列表合並,列表推導式還能否勝任呢?
當然可以,具體示例代碼如下:
>>> list01 = [1,2,3]
>>> list02 = [4,5,6]
>>> list03 = [7,8,9]
>>>
>>> [x for l in (list01, list02, list03) for x in l]
[1, 2, 3, 4, 5, 6, 7, 8, 9]
該模塊裡有一個 merge 方法,可以用於合並多個列表,如下所示
>>> list01 = [1,2,3]
>>> list02 = [4,5,6]
>>> list03 = [7,8,9]
>>>
>>> from heapq import merge
>>>
>>> list(merge(list01, list02, list03))
[1, 2, 3, 4, 5, 6, 7, 8, 9]
要注意的是,heapq.merge 除了合並多個列表外,它還會將合並後的最終的列表進行排序。
>>> list01 = [2,5,3]
>>> list02 = [1,4,6]
>>> list03 = [7,9,8]
>>>
>>> from heapq import merge
>>>
>>> list(merge(list01, list02, list03))
[1, 2, 4, 5, 3, 6, 7, 9, 8]
它的效果等價於下面這行代碼:
sorted(itertools.chain(*iterables))
如果你希望得到一個始終有序的列表,那請第一時間想到 heapq.merge,因為它采用堆排序,效率非常高。但若你不希望得到一個排過序的列表,就不要使用它了。
非常全的通俗易懂 Python 魔法方法指南(上)
非常全的通俗易懂 Python 魔法方法指南(下)
其中有一個魔法方法是 add,實際 上當我們使用第一種方法 list01 + list02 的時候,內部實際上是作用在 add 這個魔法方法上的.
所以以下兩種方法其實是等價的
>>> list01 = [1,2,3]
>>> list02 = [4,5,6]
>>>
>>> list01 + list02
[1, 2, 3, 4, 5, 6]
>>>
>>>
>>> list01.__add__(list02)
[1, 2, 3, 4, 5, 6]
借用這個魔法特性,我們可以 reduce 這個方法來對多個列表進行合並,示例代碼如下
>>> list01 = [1,2,3]
>>> list02 = [4,5,6]
>>> list03 = [7,8,9]
>>>
>>> from functools import reduce
>>> reduce(list.__add__, (list01, list02, list03))
[1, 2, 3, 4, 5, 6, 7, 8, 9]
在 yield from 後可接一個可迭代對象,用於迭代並返回其中的每一個元素。
因此,我們可以像下面這樣自定義一個合並列表的工具函數。
>>> list01 = [1,2,3]
>>> list02 = [4,5,6]
>>> list03 = [7,8,9]
>>>
>>> def merge(*lists):
... for l in lists:
... yield from l
...
>>> list(merge(list01, list02, list03))
[1, 2, 3, 4, 5, 6, 7, 8, 9]
推薦下我本人原創的 《PyCharm 中文指南》電子書,內含大量(300張)的圖解,制作之精良,值得每個 Python 工程師點個收藏。
地址是:http://pycharm.iswbm.com
————————————————
版權聲明:本文為CSDN博主「寫代碼的明哥」的原創文章,遵循CC 4.0 BY-SA版權協議,轉載請附上原文出處鏈接及本聲明。
原文鏈接:https://blog.csdn.net/weixin_36338224/article/details/109035121