Pythonのタプルリスト内の各タプルの最初の値を合計するにはどうすればよいですか?

StackOverflow https://stackoverflow.com/questions/638048

  •  10-07-2019
  •  | 
  •  

質問

次のようなタプル(常にペア)のリストがあります:

[(0, 1), (2, 3), (5, 7), (2, 1)]

各ペアの最初のアイテムの合計を検索したい、つまり:

0 + 2 + 5 + 2

Pythonでこれを行うにはどうすればよいですか?現時点では、リストを繰り返し処理しています:

sum = 0
for pair in list_of_pairs:
   sum += pair[0]

もっとPythonicな方法が必要だと感じています。

役に立ちましたか?

解決

Python 2.3と互換性のあるバージョンは

sum([pair[0] for pair in list_of_pairs])

または最近のバージョンのPythonでは、この回答またはこれ

他のヒント

sum(i for i, j in list_of_pairs)

も行います。

推奨:

sum(i for i, _ in list_of_pairs)

j の代わりに変数 _ (または gettext のエイリアスとの競合を避けるために __ を使用)少なくとも2つの利点があります:

  1. _ (プレースホルダーの略)の読みやすさ
  2. pylint は文句を言いません:"未使用の変数 'j'"

非常に大きなリストまたは多数のペアを生成するジェネレーターがある場合は、ジェネレーターベースのアプローチを使用できます。楽しみのために、 itemgetter() imap()も使用します。ただし、単純なジェネレータベースのアプローチで十分かもしれません。

import operator
import itertools

idx0 = operator.itemgetter(0)
list_of_pairs = [(0, 1), (2, 3), (5, 7), (2, 1)]
sum(itertools.imap(idx0, list_of_pairs))

itertools.imap()はPython> = 2.3で利用可能です。そこでジェネレーターベースのアプローチも使用できます。

わかりにくい(しかし楽しい)答え:

>>> sum(zip(*list_of_pairs)[0])
9

またはzipが反復可能の場合、これだけが機能するはずです:

>>> sum(zip(*list_of_pairs).__next__())
9

以下はサンプルコードです。リストの範囲を指定することもできます。

def test_lst_sum():
    lst = [1, 3, 5]
    print sum(lst)  # 9
    print sum(lst[1:])  # 8

    print sum(lst[5:])  # 0  out of range so return 0
    print sum(lst[5:-1])  # 0

    print sum(lst[1: -1])  # 3

    lst_tp = [('33', 1), ('88', 2), ('22', 3), ('44', 4)]
    print sum(x[1] for x in lst_tp[1:])  # 9

numpy配列に変換してもかまわない場合は、こちら

ライセンス: CC-BY-SA帰属
所属していません StackOverflow
scroll top