2017-05-28 139 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