Python轉換itertools.chain對象為數組的方法
更新時間：2020年02月07日 09:28:35 作者：算法猿的成長
這篇文章主要介紹了Python轉換itertools.chain對象為數組的方法,通過代碼給大家介紹了itertools 的 chain() 方法，需要的朋友可以參考下
之前做1月總結的時候說過希望每天或者每2天開始的更新一些學習筆記，這是開始的第一篇。
這篇介紹的是如何把一個 itertools.chain 對象轉換為一個數組。
參考 stackoverflow 上的一個回答：Get an array back from an itertools.chain object，鏈接如下：
https://stackoverflow.com/questions/26853860/get-an-array-back-from-an-itertools-chain-object
例子：
list_of_numbers = [[1, 2], [3], []] import itertools chain = itertools.chain(*list_of_numbers)
解決方法有兩種：
第一種比較簡單，直接采用 list 方法，如下所示：
list(chain)
但缺點有兩個：
會在外層多嵌套一個列表
效率并不高
第二個就是利用 numpy 庫的方法 np.fromiter ，示例如下：
>>> import numpy as np >>> from itertools import chain >>> list_of_numbers = [[1, 2], [3], []] >>> np.fromiter(chain(*list_of_numbers), dtype=int) array([1, 2, 3])
對比兩種方法的運算時間，如下所示：
>>> list_of_numbers = [[1, 2]*1000, [3]*1000, []]*1000 >>> %timeit np.fromiter(chain(*list_of_numbers), dtype=int) 10 loops, best of 3: 103 ms per loop >>> %timeit np.array(list(chain(*list_of_numbers))) 1 loops, best of 3: 199 ms per loop
可以看到采用 numpy 方法的運算速度會更快。
補充：下面看下itertools 的 chain() 方法
# -*- coding:utf-8 -*-
from itertools import chain
from random import randint
# 隨機生成 19 個整數（在 60 到 100 之間）
c1 = [randint(60, 100) for _ in range(19)]
# 隨機生成 24 個整數（在 60 到 100 之間）
c2 = [randint(60, 100) for _ in range(24)]
# 隨機生成 42 個整數（在 60 到 100 之間）
c3 = [randint(60, 100) for _ in range(42)]
# 隨機生成 22 個整數（在 60 到 100 之間）
c4 = [randint(60, 100) for _ in range(22)]
count = 0
# chain()可以把一組迭代對象串聯(lián)起來，形成一個更大的迭代器
for s in chain(c1, c2, c3, c4):
if s > 90:
count += 1
print('4 個班單科成績大于 90 分的人次為', count)
總結
以上所述是小編給大家介紹的Python轉換itertools.chain對象為數組的方法，希望對大家有所幫助！
相關文章
今天小編就為大家分享一篇Django xadmin開啟搜索功能的實現(xiàn)，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-11-11
這篇文章主要介紹了在類Unix系統(tǒng)上開始Python3編程入門,講解了最基礎最直觀的利用Print函數進行各種輸出的方法,需要的朋友可以參考下2015-08-08
這篇文章主要介紹了Python中的shutil標準庫用法解析,shutil模塊提供了許多關于文件和文件集合的高級操作,特別提供了支持文件復制和刪除的功能,需要的朋友可以參考下2023-09-09
動態(tài)設置django的model field的默認值操作步驟
這篇文章主要介紹了動態(tài)設置django的model field的默認值操作步驟，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-03-03
這篇文章主要介紹了Python生成器與迭代器，現(xiàn)在可以通過生成器來直接創(chuàng)建一個列表,是由于內存的限制,表的容量肯定是有限的,果我們需要一個包含幾百個元素的列表,是每次訪問的時候只訪問其中的幾個,剩下的元素不使用就很浪費內存空間,下面來了解具體內容2021-11-11
Python?Traceback(most?recent?call?last)報錯信息:示例解讀
這篇文章主要介紹了Python?Traceback(most?recent?call?last)報錯信息:示例解讀,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2023-12-12
這篇文章主要介紹了python 下載文件的多種方法匯總，幫助大家更好的理解和使用python，感興趣的朋友可以了解下2020-11-11
這篇文章主要介紹了pandas 實現(xiàn)分組后取第N行的操作，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2021-03-03
這篇文章主要為大家詳細介紹了python3射線法判斷點是否在多邊形內，具有一定的參考價值，感興趣的小伙伴們可以參考一下2019-06-06
這篇文章主要介紹了Python實現(xiàn)的排列組合、破解密碼算法,結合實例形式分析了Python排列組合、密碼破解相關數學運算操作技巧,需要的朋友可以參考下2019-04-04