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

Sponsored Link

ソースコード

#!/usr/bin/python
# coding: UTF-8

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

# 引数の前に*を付けると、関数内ではタプルとして受け取る
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])

実行結果

<type 'tuple'> (1, 2, 3)
14
<type 'tuple'> (1, 2, 3)
14
<type 'tuple'> (0, 1, 2, 3)
14

6

14

チュートリアル

• 4. More Control Flow Tools
  • 4.6 Defining Functions
    • 4.7.3 Arbitrary Argument Lists

関連すると思われる記事：

• Python: リストの初期化・出力・代入・要素数
• Python: if/for文でのin演算子の各オブジェクト毎の評価
• Ruby: lambdaメソッドを使いブロックをオブジェクト化
• Python: リストの上位型であるシーケンス型の構文 – Sequence[X:Y:Z]
• Python: リスト内包表記(リストコンプリヘンション)をfor文に書き換える

Sponsored Link