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,) кортеж целых чисел

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

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

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

likearray_like, необязательно

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

Добавлена в версии 1.20.0.

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

fromfunction: MaskedArray

Результат вызова function передаётся обратно непосредственно. Следовательно, форма fromfunction полностью определяется function. Если function возвращает скалярное значение, форма fromfunction не будет совпадать с параметром shape.

См. также

indices, meshgrid

Примечания

Ключевые слова, отличные от dtype и like, передаются в function.

Примеры

>>> np.fromfunction(lambda i, j: i, (2, 2), dtype=float)
array([[0., 0.],
       [1., 1.]])

>>> np.fromfunction(lambda i, j: j, (2, 2), dtype=float)
array([[0., 1.],
       [0., 1.]])

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