用第一个的布尔值切片两个numpy数组

Question

假设我有两个长度相同的numpy数组。

a = array([1, 0, 1, 1, 0, 1])

b = array([0, 1, 1, 1, 0, 0])

我可以对a进行切片，以便仅获得1的值。

a_slice = array([1, 1, 1, 1]).

我将如何基于a = 1的位置类似地切片b？

我会需要

b_slice = array([0, 1, 1, 0]).

Answer 1

您可以在第一个数组中切片= = 1的位置。 通过使用a == 1 。

import numpy as np

a = np.array([1, 0, 1, 1, 0, 1])
b = np.array([0, 1, 1, 1, 0, 0])

_slice = b[a == 1]
print(_slice)

您可以使用a[a == 1]来获得所有这些，但这似乎用处不大。

Answer 2

您可以将a为dtype bool

b[a.astype(bool)]
# array([0, 1, 1, 0])

Answer 3

您可以使用np.where

mask = np.where(a == 1)

a_slice = a[mask]

你得到

array([1, 1, 1, 1])


b_slice = b[mask]

你得到

array([0, 1, 1, 0])