整数数组索引

整数索引有助于基于 N 维索引来获取数组中任意元素。每个整数数组表示该维度的下标值。当索引的元素个数就是目标 ndarray 的维度时,会变得相当直接。以下示例获取了 ndarray 对象中每一行指定列的一个元素。因此,行索引包含所有行号,列索引指定要选择的元素。

以下实例获取数组中 (0, 0) 、(1, 1) 、(2, 2)位置处的元素。

也可以利用数字索引生成同样维度的数组,如下获取数组中 (0, 0) 、(0, 2) 、(2, 0)、(2, 2)位置处的元素。

布尔索引

我们可以通过一个布尔数组来索引目标数组,以此找出与布尔数组中值为True的对应的目标数组中的数据。需要注意的是,布尔数组的长度必须与目标数组对应的轴的长度一致。

如下,查看数组中大于5的元素。

如下,将数组中大于5的元素索引出来。

此外,通过布尔型数组设置值是一种常用的方式,如下,将数组大于5的值设置为5。

需要注意,Python的关键字and和or在布尔型数组中

更多推荐

python什么叫索引_python中索引是什么意思(一文详解其定义)