slicing list of lists in Python

后端 未结 6 1544
旧巷少年郎
旧巷少年郎 2020-12-10 14:40

I need to slice a list of lists in python.

A = [[1,2,3,4,5],[1,2,3,4,5],[1,2,3,4,5]]
idx = slice(0,4)
B = A[:][idx]

The code above isn\'t

6条回答
  •  夕颜
    夕颜 (楼主)
    2020-12-10 15:00

    Either:

    >>> [a[slice(0,3)] for a in A]
    [[1, 2, 3], [1, 2, 3], [1, 2, 3]]
    

    Or:

    >>> [list(filter(lambda x: x<=3, a)) for a in A]
    [[1, 2, 3], [1, 2, 3], [1, 2, 3]]
    

提交回复
热议问题