Python 擁有大量內(nèi)置函數(shù)和第三方庫，可簡化編程任務(wù)。在本文中，我們將重點介紹 Python prod() 函數(shù)。

Python math.prod() 方法用于計算可迭代對象中所有元素的積。

我們使用它來執(zhí)行數(shù)學(xué)運算，例如查找列表中所有元素的乘積。

prod()的語法格式如下：

prod(iterable, start=1)

• iterable -- 必需，可迭代對象。

• start -- 可選，指起始值。默認值為 1。

# 使用 for 循環(huán)計算
n = [1, 2, 3, 4, 5]
m = 1
for i in n:
    m *= i
print(m)

# 使用 prod() 函數(shù)計算
from math import prod
m = prod(n)
print(m)

使用 NumPy prod() 計算數(shù)組元素乘積

在 Python 中，還可以使用 NumPy 模塊的 prod() 函數(shù)來計算 NumPy 數(shù)組中所有元素的乘積。

import numpy as np
array1 = np.array([1, 2, 3, 4, 5])
m = np.prod(array1)
print(m) 

可以計算二維數(shù)組所有元素乘積，也可以使用參數(shù)axis=1計算每一行元素的乘積。

import numpy as np
array1 = np.array([[1, 2], [3, 4], [5, 6]])
m = np.prod(array1)
print(m)  # 輸出：720

import numpy as np
array1 = np.array([[1, 2], [3, 4], [5, 6]])
m = np.prod(array1,axis=1)
print(m)  # 輸出：[ 2 12 30]

使用 Pandas prod() 計算乘積

可以使用 Pandas 模塊中的 prod() 函數(shù)來計算 Pandas 數(shù)據(jù)所有元素的乘積。

import pandas as pd
 
series1 = pd.Series([1, 2, 3, 4, 5])
m = series1.prod()
print(m)  # 輸出：120

對于 Pandas 二維數(shù)據(jù)，可以計算特定軸的成績。

import pandas as pd
df1 = pd.DataFrame([[1, 2], [3, 4], [5, 6]])
row_m = df1.prod(axis=1)
col_m = df1.prod(axis=0)
print(row_m) 
print(col_m) 
# 輸出：
0     2
1    12
2    30
dtype: int64
0    15
1    48
dtype: int64