Slicing-Liste von Listen in Python

Question

Ich muss eine Liste von Listen in Python zerschneiden.

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

Der obige Code gibt mir nicht die richtige Ausgabe.

Was ich will, ist: [[1,2,3],[1,2,3],[1,2,3]]

Answer 1

Mit numpy ist es sehr einfach - man konnte nur die Scheibe durchführen:

In [1]: import numpy as np 

In [2]: A = np.array([[1,2,3,4,5],[1,2,3,4,5],[1,2,3,4,5]]) 

In [3]: A[:,:3] 
Out[3]: 
array([[1, 2, 3], 
     [1, 2, 3], 
     [1, 2, 3]]) 

Sie könnten natürlich auf die numpy.array zurückzuverwandeln list:

In [4]: A[:,:3].tolist() 
Out[4]: [[1, 2, 3], [1, 2, 3], [1, 2, 3]] 

Answer 2

Sie eine Liste Verständnis verwenden können, wie zum Beispiel: [x[0:i] for x in A] wo i 1,2,3 usw. auf, wie viele Elemente, die Sie benötigen.

Answer 3

Entweder:

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

Oder:

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

Answer 4

Sehr selten mit Segmentobjekte ist einfacher zu lesen als eine Liste Verständnis verwendet wird, und dies ist nicht einer jener Fälle.

>>> A = [[1,2,3,4,5],[1,2,3,4,5],[1,2,3,4,5]] 
>>> [sublist[:3] for sublist in A] 
[[1, 2, 3], [1, 2, 3], [1, 2, 3]] 

Dies ist sehr klar. Für jede Unterliste in A, geben Sie mir die Liste der ersten vier Elemente.