30秒学会 Python 片段 · 2017年11月9日

30秒学会 Python 片段 – similarity

Returns a list of elements that exist in both lists.

Use list comprehension on a to only keep values contained in both lists.

代码实现

def similarity(a, b):
  return [item for item in a if item in b]

使用样例

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