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]