Извлечение определенных столбцов в массиве numpy

Это простой вопрос, но скажем, что у меня есть матрица MxN. Все, что я хочу сделать, это извлечь определенные столбцы и сохранить их в другом массиве numpy, но я получаю недопустимые синтаксические ошибки. Вот код:

extractedData = data[[:,1],[:,9]]. 

Кажется, что вышеприведенная строка должна быть достаточной, но я не думаю. Я огляделся, но не смог найти ничего синтаксического мудрая относительно этого конкретного сценария.

5 Solutions collect form web for “Извлечение определенных столбцов в массиве numpy”

Я предполагаю, что вам нужны столбцы 1 и 9 ? Это

 data[:, [1, 9]] 

Предполагая, что вы хотите получить столбцы 1 и 9 с этим фрагментом кода, это должно быть:

 extractedData = data[:,[1,9]] 

если вы хотите извлечь только несколько столбцов:

 idx_IN_columns = [1, 9] extractedData = data[:,idx_IN_columns] 

если вы хотите исключить определенные столбцы:

 idx_OUT_columns = [1, 9] idx_IN_columns = [i for i in xrange(np.shape(data)[1]) if i not in idx_OUT_columns] extractedData = data[:,idx_IN_columns] 

вы можете использовать extract_data = data.ix [:, ['Column1', 'Column2']]

вы также можете использовать extractData = data ([:, 1], [:, 9])

Python - лучший язык программирования в мире.