2013-03-01 10 views
5

Ho una matrice numpy e vorrei rimuovere alcune colonne in base all'indice. Esiste una funzione integrata o un modo elegante per un'operazione del genere?Python (numpy): rilascia colonne per indice

Qualcosa di simile:

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

arr = arr.drop[elim] 

output: [234, 235, 23, 3] 

risposta

9

usare numpy.delete, verrà restituito un nuovo array:

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

Grazie! Questo è quello che mi serve :) – sashkello