Функция numpy prod() в Python с примерами

Функция numpy prod() используется в Python для вычисления произведения чисел в массиве.

• Знакомство с функцией NumPy prod()
• Примеры функции NumPy prod()
• 1) Использование функции numpy prod() в одномерном массиве
• 2) Функция numpy prod() в многомерных массивах
• 3) Выбор чисел для включения в продукт
• 4) Особые случаи

Знакомство с функцией NumPy prod()

Предположим, у вас есть три числа n, m и k. Произведение трех чисел равно nxmxk. Например, произведение 2, 3 и 4 равно 2x3x4 = 24.

Чтобы вычислить произведение чисел в массиве, используйте функцию numpy prod():

numpy.prod(a, axis=None, dtype=None, out=None, keepdims=<no value>, initial=<no value>, where=<no value>)

Примеры функции NumPy prod()

Давайте рассмотрим примеры использования функции numpy prod().

1) Использование функции numpy prod() в одномерном массиве

В следующем примере функция prod() используется для вычисления произведений чисел в одномерном массиве:

import numpy as np

a = np.arange(1, 5)
result = np.prod(a)

print(a)
print(f'result={result}')

Выход:

[1 2 3 4]
result=24

Как это работает.

• Сначала создайте массив из 4 чисел от 1 до 4, используя функцию arange().

a = np.arange(1, 5)

• Во-вторых, вычислите произведения всех чисел в массиве a:

result = np.prod(a)

• В-третьих, отобразите числа массива a и их произведение:

print(a)
print(f'result={result}')

Обратите внимание, что вы можете передать в функцию prod() объект, подобный массиву, например список.

Пример:

import numpy as np

result = np.prod([1, 2, 3, 4, 5])

print(f'result={result}')

Выход:

result=120

2) Функция numpy prod() в многомерных массивах

В следующем примере функция prod() используется для вычисления произведений всех чисел в двумерном массиве:

import numpy as np

result = np.prod([
    [1, 2],
    [3, 4]
])

print(f'result={result}')

Выход:

result=24

Чтобы вычислить произведение чисел оси, вы можете указать аргумент axis. Например, в следующем примере функция prod() используется для вычисления произведения чисел по оси 0:

import numpy as np

result = np.prod([
    [1, 2],
    [3, 4]
], axis=0)

print(f'result={result}')

Выход:

result=[3 8]

Аналогично можно вычислить произведение чисел по оси 1:

import numpy as np

result = np.prod([
    [1, 2],
    [3, 4]
], axis=1)

print(f'result={result}')

Выход:

result=[ 2 12]

3) Выбор чисел для включения в продукт

Чтобы выбрать конкретное число для включения в продукт, вы используете аргумент where. Например:

import numpy as np

a = np.array([np.nan, 3, 4])
result = np.prod(a, where=[False, True, True])
print(result)

Выход:

12.0

В этом примере массив содержит три элемента np.nan, 3 и 4.

Аргумент where использует логический список, чтобы указать, какой элемент массива a должен быть включен в продукт.

Если значение списка Where равно True, соответствующий элемент входного массива будет включен в продукт.

4) Особые случаи

Обратите внимание: если вы передаете массив целых чисел в функцию prod(), которая вызывает переполнение, prod() не выдаст ошибку. Например:

import numpy as np

result = np.prod(np.arange(1, 100))
print(f'result={result}')

Выход:

result=0

Функция prod() возвращает 1, если массив пуст. Например:

import numpy as np

result = np.prod(np.array([]))

print(f'result={result}')

Выход:

1.0