web-dev-qa-db-ja.com

Pythonのタプルのリストにタプルを追加します

私はpythonに不慣れで、これを行うための最良の方法を知りません。

ポイントを表すタプルのリストとオフセットを表す別のリストがあります。これが形成するすべての組み合わせのセットが必要です。ここにいくつかのコードがあります：

offsets = [( 0, 0),( 0,-1),( 0, 1),( 1, 0),(-1, 0)]
points = [( 1, 5),( 3, 3),( 8, 7)]

だから私の組み合わせたポイントのセットは

[( 1, 5),( 1, 4),( 1, 6),( 2, 5),( 0, 5),
 ( 3, 3),( 3, 2),( 3, 4),( 4, 3),( 2, 3),
 ( 8, 7),( 8, 6),( 8, 8),( 9, 7),( 7, 7)]

NumPyやその他のライブラリを使用できません。

pythontuples
18
rlbond
result = [(x+dx, y+dy) for x,y in points for dx,dy in offsets]

詳細は list comprehensions を参照してください。

33
Alok Singhal

ものすごく単純：

>>> rslt = []
>>> for x, y in points:
...     for dx, dy in offsets:
...         rslt.append( (x+dx, y+dy) )
... 
>>> rslt
[(1, 5), (1, 4), (1, 6), (2, 5), (0, 5), (3, 3), (3, 2), (3, 4), (4, 3), (2, 3), (8, 7), (8, 6), (8, 8), (9, 7), (7, 7)]

ポイントとオフセットを循環し、ポイントにオフセットを追加する新しいタプルを作成します。

14
Joel

個人的には、アロクの答えが好きです。ただし、 itertools のファンの場合、itertoolsベースの同等の機能（Python 2.6以降）は次のようになります）：

import itertools as it
ps = [(x+dx, y+dy) for (x, y), (dx, dy) in it.product(points, offsets)]

ただし、この場合、itertoolsソリューションは単純なソリューションよりも速くない（各x, yをアンパックする必要があるため、実際には少し遅い） Alokのシンプルなアプローチは、それぞれのx, yを1回だけアンパックしますが、すべてのオフセットに対して繰り返します）。それでも、itertools.productは、他のケースではネストされたループの優れた代替手段であるため、知っておく価値はあります！-）

8
Alex Martelli

結果の重複を気にしない場合：

result = []
for ox, oy in offsets:
    for px, py in points:
        result.append((px + ox, py + oy))

結果の重複を気にする場合：

result = set()
for ox, oy in offsets:
    for px, py in points:
        result.add((px + ox, py + oy))
5
John Machin