2014-09-30 122 views
2

this类似,我很好奇如何从numpy数组中的每个numpy数组中删除特定的元素。我的数据以下面的形式给出。我想,这应该工作:如何删除numpy数组中所有numpy数组中的第n个元素?

X = [[x1 x2 ... xn] [x1 x2 ... xn] ... [x1 x2 ... xn]] 
X.shape 
(n,|x|) 

Y=numpy.delete(X[:],1) 

我认为是Y,现在应该是:

Y = [[x1 x3 ... xn] [x1 x3 ... xn] ... [x1 x3 ... xn]] 

其中Y.shape应该等于(N-1,| Y | = | X |),但它是而不是。我没有把握什么?我的意图是能够删除X中每个数组中的所有x2(低相关变量),以发送给决策树回归器。它甚至会更好,如果我能做到这一点:

index = [ 1 3 7] 
Y=numpy.delete(X[:],index) 

这工作,如果X是不是一个“套” numpy的阵列。请参考以下链接中的回复:

array([1, 2, 3, 4, 5, 6, 7, 8, 9]) 

index 
[2, 3, 6] 

new_a = np.delete(a, index) 

new_a 
array([1, 2, 5, 6, 8, 9]) 

回答

3

您需要沿着轴应用np.delete。请参阅documentation中的第三个示例。

Y = np.delete(X, 1, axis=1) 
+0

甜!为了确保我理解轴的想法。如果我的例子是一个数组中的数组中的数组,那么我会声明axis = 2? – leonard 2014-09-30 02:24:50

+0

向下跨行(轴0),横向跨过列(轴1)。 – PlagTag 2014-09-30 14:01:56

+0

在'numpy'中,我们讨论了多维阵列,即2D或3D阵列。它们可以转换为列表(列表)的列表。但'数组数组'是不同的,只有当内部数组的大小不同时(即不规则数组)才会出现。 – hpaulj 2014-10-01 05:13:44