在每个子列表中切片一系列元素?

时间:2016-10-17 17:51:22

标签: python list list-comprehension slice

我怀疑在Python 2.7中有多种方法可以做到这一点,但我希望能够在组合中打印每个子列表的前三个元素。有没有办法在没有循环的情况下做到这一点?

combos = [ [1,2,3,.14], [5,6,7,.18], [9,10,11,.12], [1,2,3,.15] ]

这样打印语句的输出将为:

[ [1,2,3], [5,6,7], [9,10,11], [1,2,3] ]

***获得您的建议后: 我很难看到这在我的代码结构中是如何工作的,但是列表理解可以作为if语句的一部分来完成,我没有认识到:

p0combos = [ [1,2,3,.14], [5,6,7,.18], [9,10,11,.12], [1,2,3,.15] ]
p0 = [1, 2, 3]

if p0 not in [combo[:3] for combo in p0combos]:
    print combo[:3]
    print 'p0 not found'
else:
    print 'p0 found'
    print combo[3:4]

输出:

p0 found
[0.15]

谢谢大家。

3 个答案:

答案 0 :(得分:2)

[sublist[:3] for sublist in combos]

答案 1 :(得分:2)

print [temp_list[:3] for temp_list in combos]

答案 2 :(得分:0)

  

我怀疑在Python 2.7中有多种方法可以做到这一点

是的,你可以很有创意。这是另一种选择

from operator import itemgetter

map(itemgetter(slice(3)), combos)
Out[192]: [[1, 2, 3], [5, 6, 7], [9, 10, 11], [1, 2, 3]]