numpy 如何根据 boolean 条件从另一个数组中删除一个数组中的元素

Question

我有一个数组 xi

xi=
[[ 0.49671415 -0.46341769]
[-0.1382643  -0.46572975]
[ 0.64768854  0.24196227]
[ 1.52302986 -1.91328024]
[-0.23415337 -1.72491783]
[-0.23413696 -0.56228753]
[ 1.57921282 -1.01283112]
[ 0.76743473  0.31424733]
[-0.46947439 -0.90802408]
[ 0.54256004 -1.4123037 ]]

然后另一个数组 yi

[0 0 0 0 0 0 0 0 1 1 1]

如何根据 yi 中的 0 和 1 将 xi 数组分割成 2？ 我怎么能有

x1=
    [[ 0.49671415 -0.46341769]
    [-0.1382643  -0.46572975]
    [ 0.64768854  0.24196227]
    [ 1.52302986 -1.91328024]
    [-0.23415337 -1.72491783]
    [-0.23413696 -0.56228753]
    [ 1.57921282 -1.01283112]

和

x2=
    [ 0.76743473  0.31424733]
    [-0.46947439 -0.90802408]
    [ 0.54256004 -1.4123037 ]]

谢谢您的帮助！

Answer 1

您可以使用：

x1 = xi[~yi.astype(bool)]
x2 = xi[yi.astype(bool)]