Python numpy.fromiter()

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

语法

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

参数说明

• iterable：可迭代对象。
• dtype：返回数组的数据类型。
• count：要读取的项目数量。

返回值

从迭代器中返回一个新的一维 ndarray 数组。

示例

import numpy as np

# 创建迭代器
iterable = (x*x for x in range(5))

# 使用 fromiter() 函数创建数组
arr = np.fromiter(iterable, dtype=int)

print(arr)

输出结果：

[ 0  1  4  9 16]

说明

在示例中，我们首先创建一个迭代器，该迭代器生成 0 至 4 的平方数。

然后，我们使用 numpy.fromiter() 函数，将迭代器转换为新的一维数组。

在 numpy.fromiter() 函数中，我们传递了两个参数：迭代器和要返回的数组的数据类型。由于我们正在生成整数，因此我们选择 int。

最后，我们将新的一维 ndarray 数组打印为输出。

总结

numpy.fromiter() 函数是一个用于创建一维数组的有用工具，通常用于从迭代器对象中生成数组。此函数可以提高代码的效率和性能，以便在使用大量数据时能够更快地处理和存储它们。