Numpy阵列图

Question

我在缩放后将数据帧转换为numpy数组。 例如:(假设有1000个样本）

       [[ -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 ...,   

我有另一个numpy数组1D，它有1或-1（也有1000个样本）

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

如何保留样本（来自第一个数组）的样本（来自第二个数组）并删除具有-1的样本？

提前谢谢！

Answer 1

您可以使用boolean indexing

arr1[:, arr2==1]