两个列表的值的组合

Question

I am looking for an idiomatic way to combine an n-dimensional vector (given as a list) with a list of offsets, that shall be applied to every dimensions. Ie: Given I have the following values and offsets:

v = [5, 6]
o = [-1, 2, 3]

I want to obtain the following list:

n = [[4, 5], [7, 5], [8, 5], [4, 8], [7, 8], [8, 8], [4, 9], [7, 9], [8, 9]]

originating from:

n = [[5-1, 6-1], [5+2, 6-1], [5+3, 6-1], [5-1, 6+2], [5+2, 6+2], [5+3, 6+2], [5-1, 6+3], [5+2, 6+3], [5+3, 6+3]]

Performance is not an issue here and the order of the resulting list also does not matter. Any suggestions on how this can be produced without ugly nested for loops? I guess itertools provides the tools for a solution, but I did not figure it out yet.

Answer 1

from itertools import product    

[map(sum, zip(*[v, y])) for y in product(o, repeat=2)]

or, as falsetru and Dobi suggested in comments:

[map(sum, zip(v, y)) for y in product(o, repeat=len(v))]

Answer 2

itertools.product() gives you the desired combinations of o . Use that with a list comprehension to create n :

from itertools import product

n = [[v[0] + x, v[1] + y] for x, y in product(o, repeat=2)]

Demo:

>>> [[v[0] + x, v[1] + y] for x, y in product(o, repeat=2)]
[[4, 5], [4, 8], [4, 9], [7, 5], [7, 8], [7, 9], [8, 5], [8, 8], [8, 9]]

Answer 3

Use itertools.product:

>>> import itertools
>>> 
>>> v = [5, 6]
>>> o = [-1, 2, 3]
>>> 
>>> x, y = v
>>> [[x+dx, y+dy] for dx, dy in itertools.product(o, repeat=2)]
[[4, 5], [4, 8], [4, 9], [7, 5], [7, 8], [7, 9], [8, 5], [8, 8], [8, 9]]

---

originating from:

[[5-1, 6-1], [5-1, 6+2], [5-1, 6+3], [5+2, 6-1], [5+2, 6+2], [5+2, 6+3], [5+3, 6-1], [5+3, 6+2], [5+3, 6+3]]

Answer 4

您可以使用itertools模块获取所有排列。