Python转换itertools.chain对象为数组的方法
之前做1月总结的时候说过希望每天或者每2天开始的更新一些学习笔记，这是开始的第一篇。
这篇介绍的是如何把一个 itertools.chain对象转换为一个数组。
参考stackoverflow上的一个回答：Getanarraybackfromanitertools.chainobject，链接如下：
https://stackoverflow.com/questions/26853860/get-an-array-back-from-an-itertools-chain-object
例子：
list_of_numbers=[[1,2],[3],[]] importitertools chain=itertools.chain(*list_of_numbers)
解决方法有两种：
第一种比较简单，直接采用 list方法，如下所示：
list(chain)
但缺点有两个：
会在外层多嵌套一个列表
效率并不高
第二个就是利用 numpy库的方法 np.fromiter，示例如下：
>>>importnumpyasnp >>>fromitertoolsimportchain >>>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 >>>%timeitnp.fromiter(chain(*list_of_numbers),dtype=int) 10loops,bestof3:103msperloop >>>%timeitnp.array(list(chain(*list_of_numbers))) 1loops,bestof3:199msperloop
可以看到采用 numpy方法的运算速度会更快。
补充：下面看下itertools的chain()方法
#-*-coding:utf-8-*-
fromitertoolsimportchain
fromrandomimportrandint
#随机生成19个整数（在60到100之间）
c1=[randint(60,100)for_inrange(19)]
#随机生成24个整数（在60到100之间）
c2=[randint(60,100)for_inrange(24)]
#随机生成42个整数（在60到100之间）
c3=[randint(60,100)for_inrange(42)]
#随机生成22个整数（在60到100之间）
c4=[randint(60,100)for_inrange(22)]
count=0
#chain()可以把一组迭代对象串联起来，形成一个更大的迭代器
forsinchain(c1,c2,c3,c4):
ifs>90:
count+=1
print('4个班单科成绩大于90分的人次为',count)
总结
以上所述是小编给大家介绍的Python转换itertools.chain对象为数组的方法，希望对大家有所帮助！
声明：本文内容来源于网络，版权归原作者所有，内容由互联网用户自发贡献自行上传，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任。如果您发现有涉嫌版权的内容，欢迎发送邮件至：czq8825#qq.com（发邮件时，请将#更换为@）进行举报，并提供相关证据，一经查实，本站将立刻删除涉嫌侵权内容。