numpy.fromiter

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

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

Параметры

iterableiterable object

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

dtypedata-type

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

countint, optional

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

Возвращает

outndarray

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

Примечания

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

Примеры

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