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]])