python案例 - 計算一個或多個數乘積

以下函數允許計算兩個數的乘積，請稍加改造，變成可接收一個或多個數並計算乘積：

思路：利用可變參數

 

#!/usr/bin/env python3

def product(x, *y):
    res = x
    for n in y:
        res = res * n
    return res

print('product(5) =', product(5))
print('product(5, 6) =', product(5, 6))
print('product(5, 6, 7) =', product(5, 6, 7))
print('product(5, 6, 7, 9) =', product(5, 6, 7, 9))
if product(5) != 5:
    print('測試失敗!')
elif product(5, 6) != 30:
    print('測試失敗!')
elif product(5, 6, 7) != 210:
    print('測試失敗!')
elif product(5, 6, 7, 9) != 1890:
    print('測試失敗!')
else:
    try:
        product()
        print('測試失敗!')
    except TypeError:
        print('測試成功!')


# product(5) = 5
# product(5, 6) = 30
# product(5, 6, 7) = 210
# product(5, 6, 7, 9) = 1890
# 測試成功!