numpy.fromiter

原文:https://docs.scipy.org/doc/numpy/reference/generated/numpy.fromiter.html

译者:飞龙 UsyiyiCN

校对:(虚位以待)

numpy.fromiter(iterable, dtype, count=-1)

从可迭代对象创建新的1维数组。

参数:

iterable:iterable对象

为数组提供数据的可迭代对象。

dtype:数据类型

返回的数组的数据类型。

count:int,可选

要从可迭代读取的项目数。默认值为-1,表示读取所有数据。

返回:

out:ndarray

输出数组。

笔记

指定计数可提高性能。它允许fromiter预分配输出数组,而不是按需调整大小。

例子

>>> iterable = (x*x for x in range(5))
>>> np.fromiter(iterable, np.float)
array([  0.,   1.,   4.,   9.,  16.])