NUMPY065 题解：找相同元素位置

题目

找出两个数组中相同元素的索引。

解题思路

1. 读取两个数组
2. 使用 $\texttt{np.where(Z1 == Z2)}$ 找出相同元素的位置

代码

import numpy as np

n = int(input())
Z1 = np.array(list(map(int, input().split())))
Z2 = np.array(list(map(int, input().split())))
result = np.where(Z1 == Z2)
print(result[0])

代码详解

• $\texttt{Z1 == Z2}$ —— 比较两个数组，返回布尔数组
• $\texttt{np.where}$ —— 返回满足条件的索引
• $\texttt{result[0]}$ —— $\texttt{np.where}$ 返回元组，取第一个元素