gpt4 book ai didi

python - 在列表列表中查找共同元素

转载 作者:行者123 更新时间:2023-12-02 07:20:58 47 4
gpt4 key购买 nike

我有一个名为 wordlist 的单词列表列表,如下所示:

[['dog', 'cat', 'sheep', 'rabbit', 'kiss', 'time'], ['cow', 'pig', 'bomb', 'cat', 'sheep', 'cake', 'boy', 'new']]

我想找到所有子列表中的共同元素。因此,我想要的上述列表的输出应该是:

['cat', 'sheep']

为了实现这一目标,我使用以下代码创建了集合:

sets = set(tuple(row) for row in wordlist)

该集合看起来像这样:

{('cow', 'pig', 'bomb', 'cat', 'sheep', 'cake', 'boy', 'new'), ('dog', 'cat', 'sheep', 'rabbit', 'kiss', 'time')}

每个列表可以有任意数量的单词,并且可以有任意数量的列表。所以我最终可能会得到任意数量的不均匀集合。我知道我可以使用交集方法比较两个集合,但如何跨多个集合进行比较以仅返回常见项目?

最佳答案

您正在使用set错误地。您可以像这样使用它:

my_list = [['dog', 'cat', 'sheep', 'rabbit', 'kiss', 'time'], ['cow', 'pig', 'bomb', 'cat', 'sheep', 'cake', 'boy', 'new']]

# convert list of list to list of sets
my_sets = map(set, my_list)

# perform intersection on each set present in list
common_items = set.intersection(*my_sets)

这可以写成一行:

common_items = set.intersection(*map(set, my_list))

common_items 持有的值将是:

{'sheep', 'cat'}

以下解决方案使用稍微性能高效的方法给出相同的结果:

#                              v no need to type-cast sub-lists to `set` here
set(my_list[0]).intersection(*my_list[1:])

# OR,
# set(my_list[0]).intersection(*my_list)
# as intersection of set with itself returns the same set

由于 set.intersection 接受所有可迭代对象,因此无需对所有子列表进行类型转换来设置。

关于python - 在列表列表中查找共同元素,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/41925415/

47 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com