色々な方法でリストの総和を求める

はじめに

今回はPythonでリストの総和を求める

なに？組み込み関数のsum()を使えば良いって？

確かにそうだが、せっかくなので勉強のために自前で実装してみよう

以下のようなリストを考える。今回はこの総和を求めたい。

1	A = [2, 3, 5]

これが正しい方法

sum(A)

これが最も愚直な方法。

sumが存在しない言語であればこれが良い実装だろう。

def my_sum1(lst):
    sm = 0
    for e in lst:
        sm += e
    return sm
my_sum1(A)

かなり無駄な感じがしてきた。

高階関数reduceと自作の足し算をする関数addを使う。

reduceは第一引数に関数、第二引数にリストを与えることで、

リストに対して、関数の処理を繰り返して適用することができる。

from functools import reduce
def my_sum2(lst):
    def add(a, b):
        return a+b
    return reduce(add, lst)

勝手に実装にしておいて何だが、わざわざ関数addを宣言したくない。

そんな時はlambdaの出番だ。

1
2
3

from functools import reduce
def my_sum3(lst):
    return reduce(lambda a, b: a+b, lst)

再帰を使っても表現が可能だ。

先頭の要素と、残りの要素の輪を求めていく。

def my_sum4(lst):
    if len(lst)==0:
        return 0
    else:
        return lst[0] + my_sum4(lst[1:])

sumだと極端かもしれないが、標準で実装済みの処理を自分でも書いてみると良い勉強になる。