Numpy高级数组索引

NumPy 高级索引

NumPy 比一般的 Python 序列提供更多的索引方式。除了之前看到的用整数和切片的索引外，数组可以由整数数组索引、布尔索引及花式索引。

整数数组索引

对于一个二维数组 ，要获取数组中(0,0)，(1,1)和(2,0)位置处的元素。可以通过以下实例

#2021.10.23 HIT ATCI lzh
#Numpy基础知识，数组的高级索引
import numpy as np
a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print(a)
x = a[[0, 1, 2], [0, 1, 0]]#获取数组中(0, 0)、(1, 1)和(2, 0)位置处的元素
print(x)

输出：
[[1 2 3]
 [4 5 6]
 [7 8 9]]
[1 5 7]