python两数组取交集
时间: 2024-09-05 20:01:53 浏览: 18
在Python中,你可以使用集合(set)数据结构来轻松地获取两个数组的交集。集合是一种无序、不重复元素的数据结构,非常适合做这样的操作。以下是使用`intersection()`函数取两个列表(list)交集的例子:
```python
# 定义两个列表
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中,你可以使用列表推导式和`enumerate()`函数来获取两个数组的交集,并同时返回第一个数组(假设它比第二个数组长)中元素的索引。这里是一个示例:
```python
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()来实现两个数组的交集。
示例代码:
```python
a = [1, 2, 3, 4, 5, 6]
b = [4, 5, 6, 7, 8, 9]
set_a = set(a)
set_b = set(b)
intersection = set_a.intersection(set_b)
print(list(intersection))
```
输出:
```python
[4, 5, 6]
```
解析:
首先将两个数组转换成集合set类型,然后使用intersection()函数获取两个集合的交集。最后将交集转换成列表输出。