1、普通索引
普通索引是指使用单个整数或整数列表来索引数组中的元素。
1）单个元素索引
要访问 NumPy 数组的单个元素，可以使用单个整数索引。索引从 0 开始，表示数组的第一行第一个元素。
import numpy as np data = np.array([1, 2, 3, 4, 5, 6, 7, 8]) print(data[0]) # 输出：1
2）多维元素索引
import numpy as np data = np.array([[1, 2, 3, 4], [5, 6, 7, 8]]) print(data[1, 3]) #第二行第四个元素 # 输出：8
2、高级索引
高级索引允许使用布尔值或数组来索引数组中的元素。
1）布尔索引
布尔索引是一种用于根据元素的值来选择要返回的元素的索引方法。
import numpy as np # 创建一个 2D 数组 data = np.array([[1, 2, 3, 4], [5, 6, 7, 8]]) # 创建一个布尔数组 mask = data > 5 # 使用布尔索引 print(data[mask]) # 输出：[6 7 8]
2）整数数组索引
语法：arr[indices]
使用一个整数数组作为索引，可以用来选择数组中的特定元素。如二维数组操作的就是行。
import numpy as np a = np.array([[1, 2], [3, 4], [5, 6]]) idx = np.array([[0, 2],[1,2]]) print(a[idx]) # 使用整数数组索引,这是一个数组，而不能使用print(a[[0, 2],[1,2]]) # [[[1 2] # [5 6]] # [[3 4] # [5 6]]]
3）花式索引
语法：arr[ind1, ind2, ...]
花式索引指的是利用多个整数数组进行索引。花式索引根据索引数组的值作为目标数组的某个轴的下标来取值。花式索引跟切片不一样，它总是将数据复制到新数组中。如二维数组操作的是行列对应的元素。
import numpy as np
a = np.array([[1, 2], [3, 4], [5, 6]])
idx = np.array([[0, 2],[1,2],[0,1]])
#print(a[idx]) # 使用整数数组索引
#print(a)
print(a[[0,1,1]]) # 花式索引
print("---------")
print(a[[0,1,1],[0,0,0]])
#a = np.array([[[1, 2, 3], [4, 5, 6], [7, 8, 9]],
# [[10, 11, 12], [13, 14, 15], [16, 17, 18]]
# ])
#idx = np.array([[0, 1],[0,1]])
#print(a[idx]) # 使用整数数组索引3、切片索引
切片索引允许访问 NumPy 数组的连续元素子集。切片使用冒号 (:) 表示范围。切片是指使用 slice 对象来索引数组中的元素。
slice 对象由三个参数组成：start、stop 和 step。start 指定切片的起始位置，stop 指定切片的结束位置，step 指定切片的步长。
import numpy as np data = np.array([[1, 2, 3, 4], [5, 6, 7, 8]]) #访问所有行，从第2个元素开始的所有元素 print(data[:, 1:]) # 输出：[[2, 3, 4], [6, 7, 8]] # 访问从第2行开始，每行除了最后2个元素的所有元素 print(data[1:, :2]) # 输出：[[5, 6]] # 访问第一行所有元素 print(data[0, :]) # 输出：[1 2 3 4] # 访问第二行第三个元素 print(data[1, 2]) # 输出：7 # 访问第一列所有元素 print(data[:, 0]) # 输出：[1 5] # 访问前两行所有元素 print(data[:2, :]) # 输出：[[1 2 3 4] # [5 6 7 8]] # 访问从第二行开始的所有元素 print(data[1:]) # 输出：[[5 6 7 8]] # 访问前两行，从第三个元素开始的所有元素 print(data[:2, 2:]) # 输出：[[3 4] # [7 8]] # 省略号的使用 # 访问整个数组，可以使用省略号 (...) 来表示所有维度。 print(data[...]) # 输出：[[1 2 3 4] # [5 6 7 8]] #可以使用步长来指定切片的步长。 # 访问每隔一个元素 print(data[::2, ::2]) # 输出：[[1 3] # [5 7]] # 从第二行开始，每隔两个元素 print(data[1::2, ::2]) # 输出： [14 16]]