Python列表交集提取及子列表首元素匹配问题咨询

Hey there! Let's break down solutions for both of your questions clearly:

First off, your original code has a tiny syntax error—you added an extra closing square bracket! That's why it's failing. The correct set-based approach should be:

print(set(ListA).intersection(ListB))
# 或者更简洁的写法：
print(set(ListA) & set(ListB))

If sets aren't your vibe (or you need to keep duplicate elements that appear in both lists), here are two other solid approaches:

• 列表推导式（保留重复项）:

  common_elements = [x for x in ListA if x in ListB]
  print(common_elements)
  

  This keeps all instances of overlapping elements, unlike sets which automatically strip duplicates.
• 使用filter函数:

  common_elements = list(filter(lambda x: x in ListB, ListA))
  print(common_elements)
  

For this case, we first need to extract the first element from every sublist in both lists, then find their common values. Let's use your example lists to walk through it:

ListA = [[3,5], [4,4], [5,3], [6,4]]
ListB = [[3,1], [4,2], [5,2], [6,4]]

Step 1: Extract first elements from each sublist #

# 提取ListA中所有子列表的第一个元素
a_first_items = [sublist[0] for sublist in ListA]
# 提取ListB中所有子列表的第一个元素
b_first_items = [sublist[0] for sublist in ListB]

Step 2: Find common values #

• 用集合求交集（自动去重）:

  common_first_values = set(a_first_items) & set(b_first_items)
  print(common_first_values)  # 输出: {3, 4, 5, 6}
  

• 保留原顺序且去重:
  If you want to keep the order of elements as they appear in ListA (without duplicates), use dict.fromkeys():

  common_first_ordered = list(dict.fromkeys([x for x in a_first_items if x in b_first_items]))
  print(common_first_ordered)  # 输出: [3, 4, 5, 6]
  

• 保留所有匹配项（包括重复）:
  If you need to keep every instance where the first element matches (even if it repeats), skip the set/dict step:

  all_matching_first = [x for x in a_first_items if x in b_first_items]
  print(all_matching_first)  # 输出: [3, 4, 5, 6]（如果ListA有重复的第一个元素，会保留）
  

内容的提问来源于stack exchange，提问作者5Volts