NumPy 数组支持所谓的“花式索引”(Fancy Indexing),这意味着您可以使用整数列表或 NumPy 整数数组作为索引来选择数组中的元素。这种索引方式允许您从数组中选择一个不连续的元素集。
以下是一些使用列表索引 NumPy 数组的示例:
-
使用整数列表作为索引:
import numpy as np
arr = np.array([10, 20, 30, 40, 50])
indices = [1, 3, 4] # 选择索引 1, 3 和 4 的元素
print(arr[indices]) # 输出: [20, 40, 50]
-
使用布尔数组进行索引:
arr = np.array([10, 20, 30, 40, 50])
mask = [True, False, True, False, True] # 选择 True 对应的元素
print(arr[mask]) # 输出: [10, 30, 50]
-
结合条件进行索引:
arr = np.array([10, 20, 30, 40, 50]) # 选择大于 25 的元素 print(arr[arr > 25]) # 输出: [30, 40, 50]
这些花式索引技术使得 NumPy 数组在数据选择和处理方面非常灵活和强大。