python - 以下 Python 表达式如何工作？

Question

我的代码中有以下 python 表达式

import numpy as np  
a = np.array([1,0,1,0])  
b = np.array([True,False,False,True])
print a[b],b[a]  

我得到的输出看起来像这样:

[1 0] [False  True False  True]

我无法理解这个输出是如何生成的。
谁能解释一下。

Answer 1

这是两种不同的情况

a[b] 是逻辑索引。索引 b 必须是 bool 数组，与 a 相同大小。 b 的每个 bool 值表示获取/离开 a 的这个元素。即:对于向量 a, b 中 a_i, b_i 中的每一对值:如果 b_i == True，添加 a_i 到输出向量，否则忽略它。

b[a] 使用多个值进行索引。就像在普通的 python 中你可以做 b[0] 来获取 b 的第一个值，在 numpy 中你可以使用数组作为索引，来获取多个元素。在您的情况下，这会产生与 [b[1], b[0], b[1], b[0]]

相同的结果