Python numpy.fromiter()用法及代码示例

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

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

参数：

iterable:[可迭代对象]为数组提供数据的可迭代对象。

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

count:[int，可选]要读取的项目数。

Returns:[ndarray]输出数组。

代码1：

Python3

# 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：

Python3

# 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)

输出：

[ 0. 1. 4. 9. 16. 25.]

相关用法

注：本文由纯净天空筛选整理自sanjoy_62大神的英文原创作品 numpy.fromiter() function – Python。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。