Python tworzy mniejszą pod-tablicę z większej tablicy 2D NumPy?

Question

więc mam dużą tablicę NumPy że przyjmuje następującą postać:

data = [[2456447.64798471, 4, 15.717, 0.007, 5, 17.308, 0.019, 6, 13.965, 0.006], 
     [2456447.6482855, 4, 15.768, 0.018, 5, 17.347, 0.024, 6, 14.001, 0.023], 
     [2456447.648575, 4, 15.824, 0.02, 5, 17.383, 0.024, 6, 14.055, 0.023]] 

Chcę utworzyć macierz sub, który wygląda tak:

[[4, 15.717, 5, 17.308, 6, 13.965], 
[4, 15.768, 5, 17.347, 6, 14.001], 
[4, 15.824, 5, 17.383, 6, 14.055]] 

Zasadniczo chcę, aby wybrać się na pierwszą kolumnę , a następnie począwszy od czwartej kolumny chcę wybrać co 3 kolumnę. Próbowałem dowiedzieć się, jak się do tego podejść z czymś takim, jak data[1:6:?], ale nie rozumiem, jak przejść i uzyskać jedynie kolumny, które chciałem.

Potrzebuję tego również do skalowania w przypadku macierzy rozciągającej się poziomo. Więc nie chcę po prostu sztywno kodować wartości kolumn.

Answer 1

To zrobi sztuczkę, skaluje się poziomo i pionowo i jest łatwe i działa.

subArray = [] 
newRow = [] 
for row in data: 
    for i in xrange(0,len(row)): 
     if (i % 3 == 0): 
      continue 
     newRow.append(row[i]) 
    subArray.append(newRow) 
    newRow = [] 

Answer 2

Można to zrobić:

>>> data[:, [1, 2, 4, 5, 7, 8]] 
array([[ 4. , 15.717, 5. , 17.308, 6. , 13.965], 
     [ 4. , 15.768, 5. , 17.347, 6. , 14.001], 
     [ 4. , 15.824, 5. , 17.383, 6. , 14.055]])