numpy.ma.fromfunction

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

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

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

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