Python代码阅读(第37篇):获取两个列表中相同的元素

Python 代码阅读合集介绍: 为什么不推荐Python初学者直接看项目源码

本篇阅读的代码实现了返回两个列表中相同元素的列表的功能。

本篇阅读的代码片段来自于30-seconds-of-python

similarity

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

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

similarity函数接收两个列表,返回由两个列表中的相同元素组成的列表。

函数使用列表推导式,遍历所有a列表中的元素,使用in关键词判断这些元素是否在b列表中存在。

你可能感兴趣的