Pythonでインデックス値と要素の積の合計を求める方法｜enumerateの使い方

リストの各要素に対して「インデックス値 × 要素」の積を計算し、その合計を求めたい場合には、Pythonの組み込み関数 enumerate を使用するのが便利です。
本記事では、基本的なforループを使った方法と、sum() 関数を組み合わせたより簡潔な書き方の両方を紹介します。

サンプルコード

以下に具体的な実装例を示します。

my_list = [71, 23, 53, 94, 85, 26, 0, 8]

print("The list is :")
print(my_list)

my_result = 0

for index, element in enumerate(my_list):
    my_result += (index + 1) * element

print("The resultant sum is :")
print(my_result)

実行結果

The list is :
[71, 23, 53, 94, 85, 26, 0, 8]
The resultant sum is :
1297

コードの解説

• まず、整数のリストを定義し、その内容をコンソールに表示します。

• 合計を格納するための変数 my_result を 0 で初期化します。

• enumerate 関数を使うことで、リストを反復処理しながら各要素のインデックスと値を同時に取得できます。

• (index + 1) とすることでインデックスを1始まりにしています。この値を各要素と掛け合わせ、その結果を順次合計に加算していきます。

• すべての要素の処理が完了すると、最終的な合計値がコンソールに出力されます。

より簡潔な書き方：sum() とジェネレータ式

同じ処理は、sum() 関数とジェネレータ式を組み合わせることで、1行で記述することも可能です。

my_result = sum((index + 1) * element for index, element in enumerate(my_list))
print("The resultant sum is :")
print(my_result)  # 1297

この方法では、中間変数の初期化やforループが不要になるため、コードがすっきりします。計算結果は先ほどのループ版とまったく同じ 1297 になります。