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