numpy 矩阵索引|极客教程

numpy 矩阵索引

在使用 numpy 进行数据处理时，经常会涉及到对矩阵或数组的索引操作。矩阵索引是指根据一定的条件或规则，获取矩阵中的特定数据的过程。在 numpy 中，我们可以通过不同的方式进行矩阵的索引操作，包括基本索引、切片索引、花式索引等。本文将详细介绍 numpy 中矩阵索引的相关知识，并通过示例代码演示其应用。

基本索引

在 numpy 中，我们可以使用整数或布尔数组对矩阵中的元素进行基本索引。对于二维矩阵，使用 arr[i, j] 的形式可以获取第 i 行、第 j 列的元素。

import numpy as np

# 创建一个 3x3 的二维矩阵
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 获取第 0 行第 1 列的元素
element = arr[0, 1]
print(element)  # 输出 2

除了直接使用索引获取单个元素外，也可以使用切片来获取矩阵中的多个元素。切片索引的语法为 arr[start:end:step]，其中 start 表示起始位置，end 表示结束位置，step 表示步长。

import numpy as np

# 创建一个 3x3 的二维矩阵
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 获取第 0 行的所有元素
row = arr[0, :]
print(row)  # 输出 [1, 2, 3]

# 获取第 0 列的所有元素
column = arr[:, 0]
print(column)  # 输出 [1, 4, 7]

# 获取第 1 行到第 2 行（不包括第 2 行）的所有元素
submatrix = arr[1:2, :]
print(submatrix)  # 输出 [[4, 5, 6]]

花式索引

除了基本索引和切片索引外，numpy 还支持花式索引，即通过整数数组或布尔数组来获取矩阵中的元素。在花式索引中，我们可以传入一个整数数组或布尔数组，来指定获取的元素位置。

import numpy as np

# 创建一个 3x3 的二维矩阵
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 使用整数数组进行花式索引
indices = np.array([0, 2])
result = arr[indices]
print(result)  # 输出 [1, 3]

# 使用布尔数组进行花式索引
mask = arr > 5
result = arr[mask]
print(result)  # 输出 [6, 7, 8, 9]