10. 获取数组a = np.array([1,2,3,2,3,4,3,4,5,6])和数组b = np.array([7,2,10,2,7,4,9,4,9,8])元素相匹配的位置
时间: 2023-08-19 22:23:56 浏览: 112
可以使用 numpy 的 `where` 函数来获取数组中元素相匹配的位置,具体代码如下:
```python
import numpy as np
a = np.array([1, 2, 3, 2, 3, 4, 3, 4, 5, 6])
b = np.array([7, 2, 10, 2, 7, 4, 9, 4, 9, 8])
idx = np.where(a == b) # 获取元素相匹配的位置
print(idx)
```
输出结果为:
```
(array([1, 3, 5, 7]),)
```
其中,`np.where(a == b)` 表示获取数组 `a` 中与数组 `b` 相匹配的位置,返回结果是一个元组,其中第一个元素是一个一维数组,表示匹配的位置。由于这里只有一个匹配结果,因此元组中只有一个元素。
阅读全文