slicing list of lists in Python

后端 未结 6 1543
旧巷少年郎
旧巷少年郎 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条回答
  •  慢半拍i
    慢半拍i (楼主)
    2020-12-10 14:53

    A = [[1,2,3,4,5],[1,2,3,4,5],[1,2,3,4,5]]
    
    print [a[:3] for a in A]
    

    Using list comprehension

提交回复
热议问题