python : selecting row where y==1 and column is 0 in a matrix

Question

我是Python新手。这是我的代码:

np.random.seed(42)
x1=np.random.randn(5,4)
y1=np.random.randint(0,2,(5,1))
print(x1)
print(y1)

我想选择 x1 的第 1 列和 y 为 1 的行:

print(x1[y1==1, 1])

但我收到错误数组索引太多

Answer 1

您需要numpy.where提取整数数组以提供 NumPy 索引:

x1[np.where(y1==1)[0], 1]

要了解其工作原理，请注意 y1 == 1 返回以下 bool 数组:

array([[ True],
       [ True],
       [False],
       [ True],
       [False]], dtype=bool)

numpy.where 提取返回元组第一个元素中 True 元素的索引:

print(np.where(y1==1))
(array([0, 1, 3], dtype=int64), array([0, 0, 0], dtype=int64))