2012-05-16 12 views
18

Mam tablicę takiego:Wyciąganie pierwsze n kolumn macierzy numpy

array([[-0.57098887, -0.4274751 , -0.38459931, -0.58593526], 
     [-0.22279713, -0.51723555, 0.82462029, 0.05319973], 
     [ 0.67492385, -0.69294472, -0.2531966 , 0.01403201], 
     [ 0.41086611, 0.26374238, 0.32859738, -0.80848795]]) 

Teraz chcę wyodrębnić następujące elementy:

[-0.57098887, -0.4274751] 
    [-0.22279713, -0.51723555] 
    [ 0.67492385, -0.69294472] 
    [ 0.41086611, 0.26374238] 

Więc w zasadzie tylko 2 pierwszych kolumnach ..

+3

dokumentacji Numpy [krojenia] (http://docs.scipy.org/doc/numpy/reference/arrays.indexing.html#arrays-indexing). Zawsze sprawdzaj najpierw dokumentację. –

+1

@JoelCornett: Dzięki .. więc krojenie to termin .. robi się dość ciężko, jeśli znasz pojęcie, ale nie termin .. :) bardzo dziękuję :) – Fraz

Odpowiedz

42

Jeśli a to tablica:

In [11]: a[:,:2] 
Out[11]: 
array([[-0.57098887, -0.4274751 ], 
     [-0.22279713, -0.51723555], 
     [ 0.67492385, -0.69294472], 
     [ 0.41086611, 0.26374238]]) 
3

Znam to dość stare pytanie -

A = [[1,2,3],[4,5,6],[7,8,9],[10,11,12]] 

Powiedzmy, że chcemy, aby wyodrębnić - pierwsze 2 rzędy i pierwsze 3 kolumny

A_NEW = A[0:2,0:3] 
A_NEW = [[1,2,3],[4,5,6]] 

Zrozumienie składni

A_NEW = A[Start_index : stop_index, start_index : stop_index)] 

Jeśli ktoś żąda wiersza 2 i kolumny 2 i 3

A_NEW = A[1:2,1:3] 

odwołujące się do numpy indeksowania i krojenia artykułu - Indexing & Slicing

Powiązane problemy