Python numpy.fromiter()函数

numpy.fromiter()函数从一个可迭代对象中创建一个新的一维数组。

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

参数 :

iterable : [iterable object] 一个为数组提供数据的iterable对象。

dtype : [data-type] 返回数组的数据类型。

count : [int, optional] 要读取的项目数量。

返回： [ndarray] 输出数组。

代码#1：

# Python program explaining
# numpy.fromiter() function 
             
# importing numpy as geek 
import numpy as geek
  
iterable = (x * x*x for x in range(4))
  
gfg = geek.fromiter(iterable, int)
  
print (gfg)

输出 :

[ 0 1 8 27]

代码#2：

# Python program explaining
# numpy.fromiter() function 
             
# importing numpy as geek 
import numpy as geek
  
iterable = (x * x for x in range(6))
  
gfg = geek.fromiter(iterable, float)
  
print (gfg)

输出 :