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.]