1. 程式人生 > >這段程式碼很Pythonic:相見恨晚的itertools庫

這段程式碼很Pythonic:相見恨晚的itertools庫

最近事情不是很多,想寫一些技術文章分享給大家,同時也對自己一段時間來碎片化接受的知識進行一下梳理,所謂寫清楚才能說清楚,說清楚才能想清楚,就是這個道理了。

很多人都致力於把Python程式碼寫得更Pythonic,一來更符合規範且容易閱讀,二來一般Pythonic的程式碼在執行上也更有效率。今天就先給大家介紹一下Python的系統庫itertools。

itertools庫

迭代器(生成器)在Python中是一種很常用也很好用的資料結構,比起列表(list)來說,迭代器最大的優勢就是延遲計算,按需使用,從而提高開發體驗和執行效率,以至於在Python 3中map,filter等操作返回的不再是列表而是迭代器。

話雖這麼說但大家平時用到的迭代器大概只有range了,而通過iter函式把列表物件轉化為迭代器物件又有點多此一舉,這時候我們今天的主角itertools就該上場了。

使用itertools

itertools中的函式大多是返回各種迭代器物件,其中很多函式的作用我們平時要寫很多程式碼才能達到,而在執行效率上反而更低,畢竟人家是系統庫。

itertools.accumulate

簡單來說就是累加。

>> import itertools
>> x = itertools.accumulate(range(10))
>> print(list(x))
[0, 1, 3, 6, 10, 15, 21, 28, 36, 45]