30秒学会 Python 片段 – intersection

Returns a list of elements that exist in both lists.

Create a set from a and b, then use the built-in set operator & to only keep values contained in both sets, then transform the set back into a list.

代码实现

def intersection(a, b):
  _a, _b = set(a), set(b)
  return list(_a & _b)

使用样例

intersection([1, 2, 3], [4, 3, 2]) # [2, 3]