2013-03-01 4 views
5

У меня есть массив numpy и вы хотите удалить некоторые столбцы на основе индекса. Есть ли встроенная функция для нее или какой-то элегантный способ для такой операции?Python (numpy): падение столбцов по индексу

Что-то вроде:

arr = [234, 235, 23, 6, 3, 6, 23] 
elim = [3, 5, 6] 

arr = arr.drop[elim] 

output: [234, 235, 23, 3] 

ответ

9

использовать numpy.delete, он возвращает новый массив:

import numpy as np 
arr = np.array([234, 235, 23, 6, 3, 6, 23]) 
elim = [3, 5, 6] 
np.delete(arr, elim) 
+0

Спасибо! Это то, что мне нужно :) – sashkello