📜  如何从 numpy arrayt 中的 arraylist 中获取单个元素 - Python (1)

📅  最后修改于: 2023-12-03 15:08:14.415000             🧑  作者: Mango

如何从numpy array中的arraylist中获取单个元素 - Python

在numpy中,arraylist其实是一个列表,其中包含了多个numpy数组。要从arraylist中获取单个元素,可以使用索引和切片的方式进行。

1. 索引获取单个元素

要获取arraylist中的单个元素,可以使用下标索引的方式进行,如下所示:

import numpy as np

arraylist = [np.array([1,2,3]), np.array([4,5,6])]
el = arraylist[1][2] # 获取第二个数组的第三个元素
print(el)

这里我们首先定义了一个包含两个numpy数组的arraylist,然后使用arraylist[1]获取了第二个数组,再使用[2]获取了该数组的第三个元素。

输出:

6
2. 切片获取单个元素

除了使用下标索引的方式获取单个元素外,还可以使用切片的方式进行,如下所示:

import numpy as np

arraylist = [np.array([1,2,3]), np.array([4,5,6])]
el = arraylist[1][1:2] # 获取第二个数组的第二个元素
print(el)

这里我们使用了切片[1:2]来获取第二个数组的第二个元素。

输出:

[5]
3. 注意事项

需要注意的是,从arraylist中获取单个元素时,索引和切片的方式与普通的列表没有区别。但是,由于arraylist中包含的是numpy数组,因此要注意数组的维度和形状。如果多个数组的形状不一致,则无法使用相同的索引或切片方式获取元素。