2017-05-28 120 views
0

我有一个数据帧缩放后转换为numpy数组。 例子:(假设有1000个样本)numpy数组映射

 [[ -3.12819390e-01 -1.92776011e-01 -5.86198203e-01 ..., 
      3.47114065e-01 
      3.74794001e-01 2.89473684e-01] 
     [ -7.99896504e-01 -8.35721640e-01 -8.70581173e-01 ..., 
      2.31696140e-01 2.36763880e-01 1.57894737e-01] 
     [ -3.12819390e-01 -6.20289856e-02 -1.27139646e+00 ..., 

我有了1或者其他numpy的阵列1D-1(1000个样本也)

 [1 1 -1 ..., 1 -1 1] 

我如何才能让样品(来自第一个数组)有一个1(从第二个数组)并删除具有-1的样本?

预先感谢您!

回答

2

您可以使用boolean indexing

arr1[:, arr2==1] 
+0

哦,等一下,我有IndexError:指数298是出界的轴1,大小298 – hdatas

+0

@hdatas第二个数组的长度必须大于尺寸等于或小于沿着维度1的索引数组; – gzc

+0

是的,找到了pb谢谢! – hdatas