创建具有负值排列的列表

Question

谁能帮助从给定列表中创建具有负值的列表？

例如：

values = [1, 2, 3]

期望的输出是

[[-1, 2, 3], [1, -2, 3], [1, 2, -3],[-1, -2, 3], [1, -2, -3], [-1, 2, -3], [-1, -2, -3]]

Answer 1

产生 3 次 [1, -1] 的乘积并将结果与​​您的输入列表相乘：

from itertools import product

values = [1, 2, 3]
[[num * mul for num, mul in zip(values, combo)]
 for combo in product([1, -1], repeat=len(values))]

演示：

>>> from itertools import product
>>> values = [1, 2, 3]
>>> [[num * mul for num, mul in zip(values, combo)] for combo in product([1, -1], repeat=len(values))]
[[1, 2, 3], [1, 2, -3], [1, -2, 3], [1, -2, -3], [-1, 2, 3], [-1, 2, -3], [-1, -2, 3], [-1, -2, -3]]

输出顺序与您指定的略有不同，但生成的值相同，加上原始列表作为第一个元素。

Answer 2

另一种方法是喜欢这个而不是有两个列表推导

from itertools import *

print [[1*x[0], 2*x[1], 3*x[2]] for x in list(product([1,-1], repeat=3))]
[[1, 2, 3], [1, 2, -3], [1, -2, 3], [1, -2, -3], [-1, 2, 3], [-1, 2, -3], [-1, -2, 3], [-1, -2, -3]]

由于您知道列表索引处的值，您可以将它们乘以 x 中的相应索引。然而，这不像使用两个列表推导式那样灵活，比如@Martijn's answer，它可以更容易地处理可变大小的列表。