Python入门教程(四十一)Python的NumPy数组索引

以下是关于“Python入门教程(四十一)Python的NumPy数组索引”的完整攻略：

Python的NumPy数组索引

在Python的NumPy中，我们可以使用多种方法对数组进行索引。以下是常用的几种方式。

基本索引

基本索引是指使用“[ ]”进行索引，可以使用整数或布尔数组作为索引值。

整数索引

我们通常使用整数索引从数组中获取单个元素，同样可以使用整数将数组的一部分切出来。例如：

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
print(arr[0])  # 输出 1
print(arr[1:3])  # 输出 [2, 3]

布尔索引

布尔索引允许我们使用一个布尔数组作为索引值来过滤数组中的元素。例如：

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
bool_arr = np.array([True, False, True, False, False])
print(arr[bool_arr])  # 输出 [1, 3]

高级索引

高级索引是指使用整数或元组来索引数组。

整数索引

整数索引有点像使用整数值切割数组。它允许我们使用整数索引值获取数组的一部分，按照索引顺序拼接成一个新的数组。例如：

import numpy as np

arr = np.array([[1, 2], [3, 4], [5, 6]])
print(arr[[0, 1, 2], [0, 1, 0]])  # 输出 [1, 4, 5]

元组索引

元组索引是使用整数、切片和布尔值的混合形式。它允许我们获取数组的特定元素。

以下是用于元组索引的代码示例：

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print(arr[(0, 1, 2), (2, 0, 1)])  # 输出 [3, 4, 8]

其中，第一个索引值为行数组，第二个索引值为列数组。因此，该代码从第一行的第三个元素开始，取出第一行、第二行和第三行的第一个、第三个和第二个元素。

以上就是Python的NumPy数组索引的几种常用方法，希望能对你有所帮助。