numpy.fromfunction

numpy.fromfunction(function, shape, *, dtype=<class 'float'>, like=None, **kwargs)[source]

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

Полученный массив, следовательно, имеет значение 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 является плавающей точкой.

likeмассивоподобный объект, необязательно

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

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

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

fromfunctionлюбой

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