我将如何递归地从嵌套列表中获取唯一项？

Question

所以，我是递归的初学者。

我必须回答一个问题，在该问题中我定义了一个函数，该函数将由数字或字母组成的嵌套列表作为参数，并且我必须返回一个包含所有唯一值的字典。 （我必须递归解决这个问题）

def extract_unique_elements(lists):
    if isinstance(lists, int):
        return {lists}

    for i in set(lists):
        return extract_unique_elements({i})

很明显，这甚至还没有接近工作。如果是的话；但是，当传递一个嵌套列表时，它会返回如下：

>>> extract_unique_elements([1,[2,1,[5,5,[2]]]]) 
{1,2,5}

>>> extract_unique_elements([a,[a,b,[c,b,[e]]]])
{a,b,c,e}

Answer 1

这是您可以做到的一种方式：

result = set()
lists = [1,[2,1,[5,5,[2]]]]

def extract_unique_elements(lists, result):
    if isinstance(lists, list):
        for item in lists:
            extract_unique_elements(item, result)

    else:
        result.add(lists)

extract_unique_elements(lists, result)

result
# {1, 2, 5}

Answer 2

你可以使用：

def extract_unique_elements(lists):
    if isinstance(lists, int):
        return {lists}

    values = set()
    for i in lists:
        values.update(extract_unique_elements(i))

    return values

或：

def extract_unique_elements(lists):
    if isinstance(lists, int):
        return {lists}

    return {i for e in lists for i in extract_unique_elements(e)}

测试：

extract_unique_elements([1,[2,1,[5,5,[2]]]])
# {1, 2, 5}

如果您想要更紧凑的版本，您可以使用：

def extract_unique_elements(lists):
    return {i for e in lists  for i in ([e] if isinstance(e, int) else extract_unique_elements(e))}

Answer 3

或者

def myflatten(l):
    if type(l) != list:
        return([l])
    elif type(l) == list:
        if len(l) == 1:
            return(myflatten(l[0]))
        else:
            res = myflatten(l[0]) + myflatten(l[1:])
            return(res)

print(set(myflatten([1,[2,1,[5,5,[2]]]])))
#{1, 2, 5}

# OR (rectified):

def myflatten(l):
    if not isinstance(l, list):
        return [l]
    else:
        if len(l) == 1:
            return myflatten(l[0])
        else:
            res = myflatten(l[0]) + myflatten(l[1:])
            return res