-3

假设我有一个包含 5 个元素的列表,x=[a,b,c,d,e]并且我想运行一个 for 循环来打印其中两个条目比原始列表中的相应条目小 1 的所有列表。

在 Python 中执行此操作的简单方法是什么?提前致谢。

编辑:如果x=[4,5,6,7,8]我想要:

[3,4,6,7,8], [3,5,5,7,8], [3,5,6,6,8] etc.
4

2 回答 2

2

像这样的东西:

>>> from itertools import combinations
>>> lis = [0,1,2,3,4]
>>> for x,y in combinations(range(len(lis)),2):
    l = lis[:]
    l[x] -= 1
    l[y] -= 1
    print l
...     
[-1, 0, 2, 3, 4]
[-1, 1, 1, 3, 4]
[-1, 1, 2, 2, 4]
[-1, 1, 2, 3, 3]
[0, 0, 1, 3, 4]
[0, 0, 2, 2, 4]
[0, 0, 2, 3, 3]
[0, 1, 1, 2, 4]
[0, 1, 1, 3, 3]
[0, 1, 2, 2, 3]

较短的版本:

for x,y in combinations(range(len(lis)),2):
    print [item - 1 if i in (x,y) else item  for i,item in enumerate(lis)]
...     
[-1, 0, 2, 3, 4]
[-1, 1, 1, 3, 4]
[-1, 1, 2, 2, 4]
[-1, 1, 2, 3, 3]
[0, 0, 1, 3, 4]
[0, 0, 2, 2, 4]
[0, 0, 2, 3, 3]
[0, 1, 1, 2, 4]
[0, 1, 1, 3, 3]
[0, 1, 2, 2, 3]
于 2013-06-18T19:24:31.813 回答
2
from itertools import combinations
a = [1,2,3,4]
for combination in combinations(range(len(a)),r=2):
    print [c-(1 if i in combination else 0) for i,c in enumerate(a)]
于 2013-06-18T19:27:51.597 回答