17

私はいくつかの番号のリストを持っています:

[1, 2, 3, 4, 5]
[2, 3, 4, 5, 6]
[3, 4, 5, 6, 7]

使用しているすべてのリストが同じ長さであると仮定して、これらのリストの要素を追加するにはどうすればよいですか?

これを実行して上記のリストに出力したいのは、次のようなものです。

[6, 9, 12, 15, 18]

ある種のループが必要になることはわかっていますが、どうすればエレガントにループできますか?

4

6 に答える 6

32

この機能的なスタイルのコードを試してください:

>>> map(sum, zip(*lists))
[6, 9, 12, 15, 18]

このzip関数は、同じインデックスを持つ要素に一致します。

>>> zip(*lists)
[(1, 2, 3), (2, 3, 4), (3, 4, 5), (4, 5, 6), (5, 6, 7)]

次にsum、 を使用して各タプルに適用されmapます。

オンラインでの動作を確認してください: ideone


Python 3.x では、mapリストを返さないことに注意してください。リストが必要な場合は、次の質問を参照してください。

(単に呼び出すことができますlist)。

于 2012-07-01T08:25:28.940 に答える
9
>>> lis=[[1,2,3,4,5],[2,3,4,5,6],[3,4,5,6,7]]

>>> [sum(x) for x in zip(*lis)]
[6, 9, 12, 15, 18]
于 2012-07-01T08:26:14.210 に答える
6

numpy の使用:

>>> seq = np.array([
... [1,2,3,4,5],
... [2,3,4,5,6],
... [3,4,5,6,7]])
>>> np.sum(seq,axis=0)
array([ 6,  9, 12, 15, 18])
于 2012-07-01T08:51:56.293 に答える
6

仮定:

a = [1,2,3,4,5]
b = [2,3,4,5,6]
c = [3,4,5,6,7]

これを行うだけです:

[sum(n) for n in zip(*[a, b, c])]
于 2012-07-01T08:27:39.497 に答える
2

どうですか:

a = [1,2,3,4,5]
b = [2,3,4,5,6]
c = [3,4,5,6,7]

s = map(sum, zip(a,b,c))
于 2012-07-01T08:27:13.303 に答える