2013-03-01 226 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