* argsを使ってリスト要素を追加する関数を書く

Question

いくつかのベクトルのスカラ和を計算する関数を書いてみたいと思います。それは引数としていくつかのリストを取るべきです。

def add_lists(x, y, z): 
    # return sum of x, y, z 

x = add_lists([1, 2],[2, 1],[3, 1]) 
# x returns [6, 4] 

2つだけの引数のためにそれを行うのは簡単ですが、彼らが二つ以上ある場合は、私のadd_listsは[6,4]を返すようにどのように私は、*引数を使用してそれを行うことができますか？

Answer 1

あなたの質問を正しく理解している場合は、任意の量のベクトルを取り、合計を返す関数を記述したいと思いますか？関数を呼び出す際に通過すべての引数を含む - リストとしてargsを扱う、基本的に

def add_lists(*args): 
    final_vector = [0, 0] 
    for vector in args: 
     final_vector[0] += vector[0] 
     final_vector[1] += vector[1] 
    return final_vector 

：[はい、あなたはそうのような*argsを使用する必要があります。

Answer 2

あなたはzipと*argsを使用することができます。

def add_lists(*args): 
    return [sum(x) for x in zip(*args)] 

x = add_lists([1, 2],[2, 1],[3, 1]) 
# x is [6, 4] 

Answer 3

ます。また、このようにそれを行うことができます。

def add_lists(*args): 
    s1 = sum(i[0] for i in args) 
    s2 = sum(i[1] for i in args) 
    return [s1,s2] 

res = add_lists([1, 2],[2, 1],[3, 1]) 
print(res) 

出力：

[6, 4]