numpy.fromiter

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

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

Параметры

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

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

dtypeтип данных

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

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.])