Home > Archives > 2008-06-19
2008-06-19
Python: 可変個の引数を受け取る関数
- 2008-06-19 (木)
- Python
ソースコード
#!/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
チュートリアル
- Comments: 0
- Trackbacks: 0
Home > Archives > 2008-06-19