Python numpy fromiter用法及代碼示例

本文簡要介紹 python 語言中 numpy.fromiter 的用法。

用法:

numpy.fromiter(iter, dtype, count=- 1, *, like=None)

從可迭代對象創建一個新的一維數組。

參數：

iter： 可迭代對象

為數組提供數據的可迭代對象。

dtype： 數據類型

返回數組的數據類型。

count： 整數，可選

要從可迭代中讀取的項目數。默認值為 -1，表示讀取所有數據。

like： array_like

允許創建非 NumPy 數組的引用對象。如果作為 like 傳入的類似數組支持 __array_function__ 協議，則結果將由它定義。在這種情況下，它確保創建一個與通過此參數傳入的數組對象兼容的數組對象。

返回：

out： ndarray

輸出數組。

注意：

指定數數以提高性能。它允許fromiter預先分配輸出數組，而不是根據需要調整其大小。

例子：

>>> iterable = (x*x for x in range(5))
>>> np.fromiter(iterable, float)
array([  0.,   1.,   4.,   9.,  16.])