numpy.array index

在 numpy 中，可以使用整数、切片、布尔值数组和整数数组作为数组的索引。以下是一些示例：

1. 整数索引：

import numpy as np

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

2. 切片索引：

import numpy as np

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

3. 布尔值数组索引：

import numpy as np

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

4. 整数数组索引：

import numpy as np

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

需要注意的是，整数数组索引和布尔值数组索引可以用于多维数组，但是切片索引只能用于一维数组。此外，整数数组索引和布尔值数组索引可以混合使用。