如何将Python中的itertools.chain对象转换成数组？

本文共计752个文字，预计阅读时间需要4分钟。

之前做了一次月总结的时候，说过希望每天或者每两天开始更新一些学习笔记，这是第一篇。这篇介绍的是如何将一个itertools.chain对象转换为一个数组。

要将itertools.chain对象转换为数组，可以通过以下步骤实现：

1. 使用itertools.chain将多个可迭代对象连接起来。

2.使用列表推导式或循环遍历itertools.chain对象，将每个元素添加到列表中。

以下是一个示例代码：

python

from itertools import chain

假设有两个可迭代对象list1=[1, 2, 3]list2=[4, 5, 6]

使用itertools.chain连接两个列表combined=chain(list1, list2)

将itertools.chain对象转换为列表result=list(combined)

print(result) # 输出: [1, 2, 3, 4, 5, 6]

参考了Stack Overflow上的一个回答，以下是一个相关的回答链接：[Stack Overflow Answer](https://stackoverflow.com/a/6774277/1234567)。

之前做1月总结的时候说过希望每天或者每2天开始的更新一些学习笔记，这是开始的第一篇。

这篇介绍的是如何把一个itertools.chain 对象转换为一个数组。

参考 stackoverflow 上的一个回答：Get an array back from an itertools.chain object，链接如下：

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()可以把一组迭代对象串联起来，形成一个更大的迭代器 for s in chain(c1, c2, c3, c4): if s > 90: count += 1 print('4 个班单科成绩大于 90 分的人次为', count)

总结

以上所述是小编给大家介绍的Python转换itertools.chain对象为数组的方法，希望对大家有所帮助！