Python - 使用 for 循环构造集

Question

在 leetcode 上看到了这段代码，但不明白它的作用或工作原理。 Paths 是一个列表列表。

def solution(paths):
    s = set(p[0] for p in paths)    

Answer 1

好的，不确定您指的是代码的哪一部分。

def solution(**paths**):          #1
    s = set(p[0] for p in paths)  #2

#1 创建一个名为 solution 的函数，它接受一个参数(路径)
#2 为变量 分配一个集合s .
设置( ) 是一个创建集合对象的函数。
p[0] 是您想要的 forloop 项目。而你正在抢 第一个变量 在每个 集 那是在路径中
它也可以解释为:

def solution(paths):
    s = {}           # declare variable
    for item in path: # use forloop
        s.add(item[0]) #add the first element of each item in path (the set in path)
    print(s)

test_path = [ ['a','b','c'] , ['a','e','f'], ['g','h','i'] ]
solution(test_path)

请注意，在下面打印的结果中，只添加了一个“a”，因为一个集合只有唯一的变量

>>>{'g', 'a'}

您可以对列表理解进行一些研究。
我自己几个月前才开始学习 Python。
但我发现这个很棒的视频我认为它很好地解释了不同类型的理解。
Python Tutorial: Comprehensions