numpy.fromiter()函数– Python

numpy.fromiter() 函数是 NumPy 库中的一个函数，可以从一个可迭代对象中创建一个新的一维数组。该函数返回一个新的一维数组，该数组的元素来自迭代器或可迭代对象中的一个或多个元素。

这个函数的使用方式如下：

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

• iterable：可迭代对象。
• dtype：新数组的数据类型。
• count：可选参数，表示要从可迭代对象中读取的元素的数量，缺省值为-1，表示读取所有元素。

下面是一个简单的示例，展示如何使用 numpy.fromiter() 函数：

import numpy

# 定义一个可迭代对象
my_iter = (x*x for x in range(5))

# 从可迭代对象中创建一个新的一维数组
my_arr = numpy.fromiter(my_iter, int)

print(my_arr)

输出结果如下：

[ 0  1  4  9 16]

本示例中，我们创建了一个可迭代对象 my_iter，它包含了前 5 个自然数的平方，然后调用 numpy.fromiter() 函数，使用 my_iter 作为输入参数创建了一个新的一维数组 my_arr，最后输出 my_arr。

在上面的示例中，我们显式地指定了新数组的数据类型为 int，如果不显式指定，numpy.fromiter() 函数会尝试自动推断数据类型。

总之， numpy.fromiter() 函数是一个用于从可迭代对象中创建一维数组的非常有用的函数，能够帮助程序员轻松、高效地处理数据，提高开发效率。