index函数可以直接获取列表中某个元素的索引,但仅限一维列表,想获取多维列表中某个元素的索引有什么简单快捷的方法吗
时间: 2024-04-17 14:26:34 浏览: 136
是的,如果你想要获取多维列表中某个元素的索引,可以使用NumPy库中的`np.where()`函数来实现。这个函数可以在多维数组中查找满足条件的元素,并返回它们的索引。
下面是一个示例代码,演示了如何使用`np.where()`函数来获取多维列表中某个元素的索引:
```python
import numpy as np
# 创建一个多维列表
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 获取元素5的索引
indices = np.where(arr == 5)
# 打印索引
print(indices)
```
输出结果为:
```
(array([1]), array([1]))
```
这里的输出结果表示元素5在多维列表中的索引位置为[1, 1]。
你也可以使用`np.argwhere()`函数来实现类似的功能,它返回满足条件的元素的坐标列表。示例代码如下:
```python
import numpy as np
# 创建一个多维列表
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 获取元素5的索引
indices = np.argwhere(arr == 5)
# 打印索引
print(indices)
```
输出结果为:
```
[[1 1]]
```
这里的输出结果也表示元素5在多维列表中的索引位置为[1, 1]。
阅读全文