`numpy.prod()`

Функция numpy.prod() (старое имя numpy.product) перемножает все элементы массива или вдоль указанной оси. Часто используется в комбинаторике, теории вероятностей и вычислении факториалов.

Сигнатура

numpy.prod(a, axis=None, dtype=None, out=None, keepdims=False, initial=1, where=True)

Параметры

a — входной массив.
axis — ось вычисления (None — по всем элементам).
dtype — тип данных результата.
initial — стартовое значение (по умолчанию 1).
where — маска включения элементов.

Возвращает

Скаляр или ndarray произведений.

Пример 1. Произведение всех элементов

import numpy as np

a = np.array([1, 2, 3, 4])
print(np.prod(a))

Это факториал 4!.

Пример 2. Произведение по оси

import numpy as np

data = np.array([[1, 2, 3],
                 [4, 5, 6]])
print(np.prod(data, axis=0))
print(np.prod(data, axis=1))

[ 4 10 18]
[  6 120]

Пример 3. Объём из формы массива

import numpy as np

arr = np.zeros((4, 3, 2))
total = np.prod(arr.shape)
print(f"Элементов: {total}")

Элементов: 24

Пример 4. С тип данных int64 (для больших чисел)

import numpy as np

nums = np.arange(1, 11)
print(np.prod(nums, dtype=np.int64))

10! = 3 628 800.

Пример 5. Произведение только положительных

import numpy as np

a = np.array([2, -3, 4, 0, 5])
print(np.prod(a, where=(a > 0)))

Учтены только 2, 4, 5.

См. также

Примечание

Лицензия и источники

Техническое описание функции адаптировано из официальной документации NumPy (https://numpy.org/doc/stable/), BSD-3-Clause License. Примеры и пояснения — © AlashEd Wiki.

numpy.prod()