NumPy 函数

NumPy fromiter() 函数用于从可迭代对象创建新的一维数组。使用该函数的语法如下:

语法

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

参数

iterable必填。 指定为数组提供数据的可迭代对象。
dtype必需。 指定返回数组所需的数据类型。
count可选。 指定要从可迭代对象中读取的项目数。默认值为-1,表示读取所有数据。

返回值

返回从可迭代对象创建的数组。

示例:

在下面的示例中,fromiter() 函数用于从可迭代对象创建 numpy 数组。

import numpy as np

it = (x*x for x in range(5))

#从可迭代创建 numpy 数组
Arr = np.fromiter(it, dtype=float)
print("Arr is:", Arr) 

上述代码的输出将是:

Arr is: [ 0.  1.  4.  9. 16.]