Skip to main content
 首页 » 编程设计

python之从 itertools.chain 对象获取一个数组

2025年05月04日15kuangbin

假设我有 list_of_numbers = [[1, 2], [3], []]我想要更简单的对象列表对象 x = [1, 2, 3] .

遵循 this related solution 的逻辑, 我愿意

list_of_numbers = [[1, 2], [3], []] 
import itertools 
chain = itertools.chain(*list_of_numbers) 

不幸的是,chain不是我想要的,因为(例如)运行 chain在控制台返回 <itertools.chain object at 0x7fb535e17790> .

函数是什么f这样如果我做 x = f(chain)然后输入 x在控制台我得到 [1, 2, 3] ?

更新:其实我最终需要的结果是array([1, 2, 3]) .我正在对所选答案的评论中添加一行来解决这个问题。

请您参考如下方法:

列表。如果你执行 list(chain) 它应该可以工作。但仅将其用于调试目的,通常效率不高。