numpy.fromiter

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

Создает новый одномерный массив из итерируемого объекта.

Параметры:

iterитерируемый объект

Итерируемый объект, предоставляющий данные для массива.

dtypeтип данных

Тип данных возвращаемого массива.

Изменено в версии 1.23: Теперь поддерживаются типы данных объектов и подмассивов (обратите внимание, что окончательный результат не является одномерным для типа данных подмассива).

countцелое число, необязательно

Количество элементов для чтения из iterable. По умолчанию -1, что означает чтение всех данных.

likeмассив_подобный объект, необязательно

Объект-ссылка для создания массивов, которые не являются массивами NumPy. Если в качестве like передан массив-подобный объект, поддерживающий протокол __array_function__, результат будет определяться им. В этом случае он гарантирует создание объекта массива, совместимого с переданным через этот аргумент.

Добавлена в версии 1.20.0.

Возвращает:

outndarray

Массив вывода.

Примечания

Укажите count для повышения производительности. Это позволяет fromiter предварительно выделить массив вывода, вместо изменения его размера по мере необходимости.

Примеры

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

Тщательно сконструированный тип данных подмассива приведет к результату большей размерности:

>>> iterable = ((x+1, x+2) for x in range(5))
>>> np.fromiter(iterable, dtype=np.dtype((int, 2)))
array([[1, 2],
       [2, 3],
       [3, 4],
       [4, 5],
       [5, 6]])