如何获得numpy.array数组的索引
时间: 2024-10-27 21:06:08 浏览: 26
在NumPy中,你可以使用`numpy.where()`函数或者简单的整数索引来获取数组的索引。这两个方法都可以用于一维数组,但对于二维或多维数组,它们有不同的用法。
1. **使用**`numpy.where()`**:**
这个函数返回的是满足条件的元素的索引。例如,获取所有大于某个值的元素的索引:
```python
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
indices = np.where(arr > 3)
print(indices) # 输出 ((3,), (4,))
```
如果你想获取特定值的所有索引,可以这样做:
```python
indices = np.where(arr == 4)
print(indices) # 输出 ((3,))
```
2. **简单整数索引:**
对于一维数组,直接使用整数索引即可。例如,`arr[0]`表示第一个元素的索引,`arr[-1]`表示最后一个元素的索引。对于二维数组,可以提供两个索引,如`arr[i, j]`。
对于多维数组,你需要使用`[]`操作符并提供一系列的索引,例如获取行索引为2、列索引为3的元素:
```python
multi_dim_arr = np.array([[1, 2], [3, 4], [5, 6]])
index = (2, 3)
element = multi_dim_arr[index]
print(element) # 输出 6
```
阅读全文