Python | numpy ndarray.item()

简介

numpy.ndarray.item() 函数返回数组中单个元素。

如果数组具有更多维度，则需要使用下标来访问其元素。item() 以整数作为下标，并返回该下标位置的值。

语法

numpy.ndarray.item(index)

参数

• index：选取元素的下标。

返回值

函数返回数组中单个元素。

如果数组具有更多维度，则需要使用下标来访问其元素。item() 以整数作为下标，并返回该下标位置的值。

示例

import numpy as np

# 创建一个二维数组
a = np.array([[1,2], [3,4]])

# 输出数组中第1行第2个元素
print("第1行第2个元素为：", a.item(2))

输出结果为：

第1行第2个元素为： 2

在示例中，我们首先引入了numpy库，然后创建了一个二维数组a，接着使用item() 方法输出了a数组中的第1行第2个元素。

注意事项

• 如果下标超出了数组维度，则会抛出一个IndexError异常。

import numpy as np

# 创建一个二维数组（2 X 2）
a = np.array([[1,2], [3,4]])

# 创建一个三维数组（2 X 2 X 2）
b = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])

# 输出数组中不存在的元素
print(a.item(3))      # IndexError: index 3 is out of bounds for axis 0 with size 2
print(b.item(8))      # IndexError: index 8 is out of bounds for axis 0 with size 2

在该示例中，我们为了说明超出下标的情况，创建的了两个数组（一个二维数组和一个三维数组）。接着使用**item()**访问数组中不存在的元素。由于数组并不存在这个元素，因此程序会抛出一个IndexError异常。