什么是最好的方法来打开一个冻集列表？

Question

不允许使用任何导入。

输入：

[frozenset({1}), frozenset({32}), frozenset({40}), frozenset(), frozenset({76})]

所需输出：

[1,32,40,76]

我在找一个简洁的方法。我知道如何用多行for循环来解包它。

Answer 1

我们可以在frozenset的列表中迭代，并将它们的数据类型更改为list（请参阅此处）

l = [frozenset({1}), frozenset({32}), frozenset({40}), frozenset(), frozenset({76})]
arr = []
for element in l:
    new = list(element)
    if new != []:
        arr.append(new[0])
print(arr)

Answer 2

您可以使用列表理解来生成输出。为了提取每个frozenset中的值，您可以在其上创建一个迭代器，并使用迭代器的next()方法获取第一个且唯一的值，如果它不为空：

lst =  [frozenset({1}), frozenset({32}), frozenset({40}), frozenset(), frozenset({76})]

out = [next(iter(fset)) for fset in lst if fset]

print(out)
# [1, 32, 40, 76]