numpy.ma.fromfunction

numpy.ma.fromfunction(function, shape, **kwargs) = <numpy.ma.core._convert2ma object>

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

Результат — массив со значением 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,) кортеж целых чисел Форма выходного массива, которая также определяет форму массивов координат, передаваемых в 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]])

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