numpy.ma.fromfunction

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.

likearray_like

Объект-ссылка, позволяющий создавать массивы, которые не являются массивами NumPy. Если массив-подобный объект, переданный в like поддерживает протокол __array_function__, результат будет определяться им. В этом случае это гарантирует создание объекта массива, совместимого с переданным через этот аргумент.

Примечание

Ключевое слово like — экспериментальная функция, ожидающая принятия NEP 35.

Новая в версии 1.20.0.

Возвращаемое значение

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