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 — float.

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