使用numpy创建下列数组[1,1,2,3,4,5,6,1,1,6],找出数组中1元素的位置索引并输出索引(使用两种方法)。
时间: 2023-10-02 12:10:12 浏览: 63
方法一:
```python
import numpy as np
arr = np.array([1, 1, 2, 3, 4, 5, 6, 1, 1, 6])
# 使用 np.where() 函数查找元素 1 的位置索引
indexes = np.where(arr == 1)[0]
print("元素 1 的位置索引为:", indexes)
```
输出结果为:
```
元素 1 的位置索引为: [0 1 7 8]
```
方法二:
```python
import numpy as np
arr = np.array([1, 1, 2, 3, 4, 5, 6, 1, 1, 6])
# 使用 np.argwhere() 函数查找元素 1 的位置索引
indexes = np.argwhere(arr == 1).flatten()
print("元素 1 的位置索引为:", indexes)
```
输出结果为:
```
元素 1 的位置索引为: [0 1 7 8]
```
阅读全文