numpy.fromfunction

numpy.fromfunction(function, shape, **kwargs) [source]

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

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

Параметры:

function : callable Функция вызывается с N параметрами, где N — ранг shape. Каждый параметр представляет координаты массива, изменяющиеся вдоль определённой оси. Например, если shape было (2, 2), то параметрами были бы array([[0, 0], [1, 1]]) и array([[0, 1], [0, 1]]) shape : (N,) tuple of ints Форма выходного массива, которая также определяет форму массивов координат, передаваемых функции function. dtype : data-type, optional Тип данных массивов координат, передаваемых функции function. По умолчанию dtype — float.

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

fromfunction : any Результат вызова функции 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]])