详解NumPy中数组的索引和取值

目录

• NumPy数组索引
  • 单个元素索引
  • 多维数组索引
  • 数组切片索引
• NumPy数组取值
  • 获取单个元素的值
  • 获取多个元素的值

在NumPy中，可以使用索引和切片操作来获取数组中的元素和子数组。下面详细介绍NumPy数组的索引和取值方法。

NumPy数组索引

NumPy数组可以像Python列表一样使用索引来获取元素。数组的索引从0开始，可以是负数，表示从末尾开始索引。可以使用以下方法对NumPy数组进行索引：

单个元素索引

可以通过指定元素的下标来获取数组中的单个元素，如：

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
print(arr[0])  # 输出第一个元素 1
print(arr[-1])  # 输出最后一个元素 5

多维数组索引

可以通过指定元素在多维数组中的位置来获取元素，如：

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print(arr[0, 0])  # 输出第一个元素 1
print(arr[1, 1])  # 输出中间的元素 5
print(arr[2, -1])  # 输出最后一个元素 9

数组切片索引

可以使用切片语法来获取数组的子数组。可以指定切片的开始、结束和步长，如：

import numpy as np

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

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

NumPy数组取值

NumPy数组中的元素类型是固定的，可以使用不同的方法来获取它们的值。

获取整个数组的值
可以使用print()函数来输出整个数组的值，如：

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
print(arr)  # 输出整个数组 [1 2 3 4 5]

获取单个元素的值

可以使用索引来获取数组中单个元素的值，如：

import numpy as np

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

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print(arr[1, 1])  # 输出中间的元素 5

获取多个元素的值

可以使用切片来获取数组中多个元素的值，如：

import numpy as np

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

以上就是本文的全部内容。