numpy.fromiter

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

Функция numpy.fromiter() создает одномерный массив из итерируемого объекта.

Параметры:
iterable - итерируемый объект.
Объект, который поддерживает интерфейс итераций.
dtype - тип данных NumPy (необязательный)
Определяет тип данных выходного массива.
count - целое число (необязательный)
Количество считываемых элементов из объекта. По умолчанию count=-1, что означает чтение всех элементов.
Возвращает:
результат - массив NumPy
Одномерный массив значений из указанного итерируемого объекта.
Смотрите так же: frombuffer, fromfile, fromstring, fromfunction

Замечание

Если количество элементов в итерируемом объекте известно, то указав его в параметре count, можно немного повысить производительность функции fromiter. данный параметр позволяет сразу создать выходной массив и затем заполнять его, а не изменять его каждый раз с добавлением нового элемента.

Примеры

>>> import numpy as np
>>> 
>>> a = range(10)
>>> 
>>> np.fromiter(a, dtype = int)
array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
>>> 
>>> 
>>> b = [2*x + 1 for x in range(10)]
>>> b
[1, 3, 5, 7, 9, 11, 13, 15, 17, 19]
>>> 
>>> np.fromiter(b, dtype = int)
array([ 1,  3,  5,  7,  9, 11, 13, 15, 17, 19])