numpy.fromfunction

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

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

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

Параметры:

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

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

fromfunction : любой Результат вызова 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]], dtype=bool)

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