numpy.ma.fromfunction

numpy.ma.fromfunction(function, shape, **kwargs) = <numpy.ma.core._convert2ma instance at 0x52d1d0cc>

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

Полученный массив, следовательно, имеет значение 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]])