numpy.ma.fromfunction

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

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

Полученный массив, следовательно, имеет значение fn(x, y, z) в координате (x, y, z).

Параметры

functioncallable

Функция вызывается с N параметрами, где N — ранг shape. Каждый параметр представляет координаты массива, меняющиеся вдоль конкретной оси. Например, если shape был бы (2, 2), то параметрами были бы array([[0, 0], [1, 1]]) и array([[0, 1], [0, 1]])

shape(N,) tuple of ints

Форма выходного массива, которая также определяет форму массивов координат, передаваемых функции function.

dtypedata-type, optional

Тип данных массивов координат, передаваемых функции function. По умолчанию, dtype — float.

Возвращает

fromfunctionany

Результат вызова функции 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]])