Python: 可変個の引数を受け取る関数

ソースコード

# 可変個の引数を受け取る関数
# 引数の前に*を付けると、関数内ではタプルとして受け取る
def sigma_sq(*nums): # Σf(k)^xのを計算、f(n)はnumsの要素
    print (type(nums), nums)
    x = 2 # 乗数
    n_elem = len(nums) # タプルの長さ
    sum = 0
    for elem in nums:
        sum += pow(elem, x) # x乗する
    return sum

print(sigma_sq(1, 2, 3))
# 関数呼び出し側の引数: リスト[タプル]の前に*を付けると
# 関数側でリスト[タプル]を展開し、個々の要素を埋め込んでくれる
list = [1, 2, 3]
print(sigma_sq(*list))
print(sigma_sq(*[x for x in range(4)]))
def plusx3(a, b, c):
    return a+b+c
print ('n', plusx3(*list))
# 補足: 上の処理を簡潔に書く
sq = 2
print ('n', sum([pow(x, sq) for x in list]))

実行結果

 (1, 2, 3)
14
 (1, 2, 3)
14
 (0, 1, 2, 3)
14
n 6
n 14

チュートリアル

• 4. More Control Flow Tools