上QQ阅读APP看书,第一时间看更新
4.1.3 花式索引
花式索引指的是利用整数数组进行索引,首先创建一个服从标准正态分布的数组,示例代码如下:
下面依次按照第3行、第4行、第5行、第2行提取数据(注意默认索引是从0开始,所以索引2对应的是第3行,索引3对应的是第4行,索引4对应的是第5行,索引1对应的是第2行),示例代码和输出如下:
花式索引的结果与普通索引是一致的,只不过,花式索引简化了索引过程,而且实现了按一定的顺序排列。
此外,还可以使用负数进行索引,示例代码和输出如下:
如果一次传入两个索引数组,就会返回一个一维数组,其中的元素对应各个索引元组,示例代码如下:
print(a[[1,3,2,4],[2,0,4,4]])
[-0.88364558 1.7644797 -0.2399192 -0.59371321]
对于多维数组,花式索引也是适用的,例如创建一个3×3×3维的数组,示例代码和输出如下:
下面再传入多个索引数组,也会返回一个一维数组,示例代码和输出如下:
print(b[[1,2],[0,1],[2,2]])
[11 23]