Python列表理解返回列表的边缘值
如果我有一个python列表,例如:Python列表理解返回列表的边缘值,python,data-structures,iteration,list-comprehension,Python,Data Structures,Iteration,List Comprehension,如果我有一个python列表,例如: stuff = [1, 2, 3, 4, 5, 6, 7, 8, 9] 长度为n(在本例中为9),我对创建长度为n/2(在本例中为4)的列表感兴趣。我想要原始列表中所有可能的n/2值集,例如: [1, 2, 3, 4], [2, 3, 4, 5], ..., [9, 1, 2, 3] 是否有一些列表理解代码可以用来遍历列表并检索所有这些子列表?我不关心列表中值的顺序,我只是想找到一种生成列表的聪明方法。您需要的是 (编辑:如果顺序很重要,则使用排列)
stuff = [1, 2, 3, 4, 5, 6, 7, 8, 9]
长度为n(在本例中为9),我对创建长度为n/2(在本例中为4)的列表感兴趣。我想要原始列表中所有可能的n/2值集,例如:
[1, 2, 3, 4], [2, 3, 4, 5], ..., [9, 1, 2, 3]
是否有一些列表理解代码可以用来遍历列表并检索所有这些子列表?我不关心列表中值的顺序,我只是想找到一种生成列表的聪明方法。您需要的是
(编辑:如果顺序很重要,则使用排列)
请注意,此函数在Python 2.5中不可用。在这种情况下,您可以从上述链接复制代码:
def combinations(iterable, r):
# combinations('ABCD', 2) --> AB AC AD BC BD CD
# combinations(range(4), 3) --> 012 013 023 123
pool = tuple(iterable)
n = len(pool)
if r > n:
return
indices = range(r)
yield tuple(pool[i] for i in indices)
while True:
for i in reversed(range(r)):
if indices[i] != i + n - r:
break
else:
return
indices[i] += 1
for j in range(i+1, r):
indices[j] = indices[j-1] + 1
yield tuple(pool[i] for i in indices)
然后
stuff = range(9)
what_i_want = [i for i in combinations(stuff, len(stuff)/2)]
使用或(取决于您是否希望,例如,[1,2,3,4]
和[4,3,2,1]
与可选的第二个参数一起指定长度
stuff = [1, 2, 3, 4, 5, 6, 7, 8, 9]
itertools.permutations(stuff, 4) # will return all possible lists of length 4
itertools.combinations(stuff, 4) # will return all possible choices of 4 elements
这是假设您不仅需要连续的元素
更新
因为您指定不关心顺序,所以您可能要查找的是itertools.combines()
注意:以上代码基于示例中的假设
[1, 2, 3, 4], [2, 3, 4, 5], ..., [9, 1, 2, 3]
如果您确实需要所有可能的值,则需要使用itertools。置换或组合功能,如其他人所建议的。数字列表只是一个示例。比如说,这段代码适用于字符串列表吗?我想我还不清楚。我只对由原始列表中的连续值组成的子列表感兴趣,包括换行符。
[1, 2, 3, 4], [2, 3, 4, 5], ..., [9, 1, 2, 3]