numpy.fromfunction

numpy.fromfunction(function, shape, *, dtype=<class 'float'>, **kwargs) [source]

Создаёт массив путём выполнения функции по каждой координате.

Получающийся массив, следовательно, имеет значение fn(x, y, z) в координате (x, y, z).

Параметры

functioncallable

Функция вызывается с N параметрами, где N — ранг shape. Каждый параметр представляет координаты массива, изменяющиеся вдоль определённой оси. Например, если shape было (2, 2), то параметрами были бы array([[0, 0], [1, 1]]) и array([[0, 1], [0, 1]])

shape(N,) tuple of ints

Форма выходного массива, которая также определяет форму массивов координат, передаваемых функции function.

dtypedata-type, optional

Тип данных массивов координат, передаваемых функции function. По умолчанию, dtype — float.

Возвращаемое значение

fromfunctionany

Результат вызова функции function возвращается напрямую. Таким образом, форма массива fromfunction полностью определяется параметром function. Если функция function возвращает скалярное значение, форма массива fromfunction не будет соответствовать параметру shape.

См. также

indices, meshgrid

Примечания

Ключевые слова, отличные от dtype, передаются функции function.

Примеры

>>> np.fromfunction(lambda i, j: i == j, (3, 3), dtype=int)
array([[ True, False, False],
       [False,  True, False],
       [False, False,  True]])

>>> np.fromfunction(lambda i, j: i + j, (3, 3), dtype=int)
array([[0, 1, 2],
       [1, 2, 3],
       [2, 3, 4]])