numpy.ma.fromfunction

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

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

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

Параметры

functionвызываемый объект

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

shape(N,) кортеж целых чисел

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

dtypeтип данных, необязательно

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

likearray_like

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

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

Возвращает

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

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