python两数组取交集并返回第一个数组索引

在Python中，你可以使用列表推导式和`enumerate()`函数来获取两个数组的交集，并同时返回第一个数组（假设它比第二个数组长）中元素的索引。这里是一个示例：

def find_intersection_indices(array1, array2):
    intersection = [i for i, x in enumerate(array1) if x in array2]
    return intersection

# 示例
array1 = [1, 2, 3, 4, 5]
array2 = [3, 4, 5, 6, 7]
result = find_intersection_indices(array1, array2)
print("交集元素及在第一个数组的索引:", result)

相关问题

python两数组取交集

在Python中，你可以使用集合（set）数据结构来轻松地获取两个数组的交集。集合是一种无序、不重复元素的数据结构，非常适合做这样的操作。以下是使用`intersection()`函数取两个列表（list）交集的例子：

# 定义两个列表
list1 = [1, 2, 3, 4, 5]
list2 = [4, 5, 6, 7, 8]

# 将列表转换为集合
set1 = set(list1)
set2 = set(list2)

# 使用intersection()获取交集并返回结果为一个新的集合
intersection_set = set1.intersection(set2)

# 如果需要将交集转换回列表，可以这样做
intersection_list = list(intersection_set)

print("两个列表的交集：", intersection_list)

python二维数组交集并返回序号

问题是给定两个二维数组，需要找到它们的交集并返回交集中元素在两个数组中的序号。

我们可以使用嵌套循环来解决这个问题。首先，我们遍历第一个二维数组的每一个元素，在内层循环中，我们再遍历第二个二维数组的每一个元素。

当两个元素相等时，我们就找到了一个交集元素。我们可以记录该元素在两个数组中的序号，并将其保存在一个结果列表中。

以下是使用Python代码实现该算法的例子：

def find_intersection(array1, array2):
    result = []
    for i in range(len(array1)):
        for j in range(len(array2)):
            if array1[i] == array2[j]:
                result.append([i, j])
    return result

# 测试
array1 = [[1, 2], [3, 4], [5, 6]]
array2 = [[3, 4], [7, 8], [9, 10]]
intersection = find_intersection(array1, array2)
print(intersection)

输出结果为：[[1, 0]]

在这个例子中，第一个二维数组是`[[1, 2], [3, 4], [5, 6]]`，第二个二维数组是`[[3, 4], [7, 8], [9, 10]]`。它们的交集是`[[3, 4]]`，在第一个数组中的序号是1，在第二个数组中的序号是0。所以最后返回结果为`[[1, 0]]`。

请注意，这只是一种解决该问题的方法，还有其他更高效的算法可以使用，但是该方法在解决问题上是有效的。