2013-03-01 90 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 2013-03-01 03:20:39