之前做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,[]]
>>> 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()可以把一組迭代物件串聯起來，形成一個更大的迭代器
for s in chain(c1,c2,c3,c4):
  if s > 90:
    count += 1
print('4 個班單科成績大於 90 分的人次為',count)

總結

以上所述是小編給大家介紹的Python轉換itertools.chain物件為陣列的方法，希望對大家有所幫助！