numpy.fromfunction

numpy.fromfunction(function, shape, **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]])