Question

我正在尝试创建一个函数，该函数将嵌套列表和项目作为输入，并返回索引列表。例如list = [0, 5, [6, 8, [7, 3, 6]], 9, 10]和item = 7应该返回[2, 2, 0]，因为list[2][2][0] = 7

我的代码应该可以正常工作，因为我可以打印所需的输出，但是当我运行它时，它返回None。

def find_item(list_input, item, current):
    if list_input == item:
        print(current) ## this does give the desires output, but the function should return not print
        return current
    else:
        if isinstance(list_input, list):
            for j in range(len(list_input)):
                current.append(j)
                find_item(list_input[j], item, current)
                del current[-1]

我在这里俯瞰什么？

Answer 1

如@tzaman所述，您需要处理find_item递归调用的返回值。如果递归调用的返回值是一个列表，则意味着找到了搜索到的项目，我们需要停止递归。

以下修改将返回搜索到的项目的最早发现索引。如果未找到任何项目，它将返回None。

def find_item(list_input, item, current):
    if list_input == item:
        return current
    else:
        if isinstance(list_input, list):
            for j in range(len(list_input)):
                current.append(j)
                search_result = find_item(list_input[j], item, current)
                if isinstance(search_result, list):
                    return search_result
                del current[-1]

list_input  = [0, 5, [6, 8, [7, 3, 6]], 9, 10]
item = 7
print(find_item(list_input, item, []))

list_input  = [0, 5, [6, 8, [7, 3, 6]], 9, 10]
item = 9
print(find_item(list_input, item, []))

list_input  = [0, 5, [6, 8, [7, 3, 6]], [30, 4], 9, 10]
item = 4
print(find_item(list_input, item, []))

list_input  = [0, 5, [6, 8, [7, 3, 6]], [30, 4], 9, 10]
item = 400
print(find_item(list_input, item, []))

输出：

[2, 2, 0]
[3]
[3, 1]
None

Answer 2

当有人在递归函数的中间插入for循环时，它总是让我感到困扰！这是解决此问题的另一种方法：

def find_item(list_input, item, index=0):
    if list_input:
        head, *tail = list_input

        if head == item:
            return [index]

        if isinstance(head, list):
            if result := find_item(head, item):
                return [index] + result

        return find_item(tail, item, index + 1)
    
    return list_input

list_input  = [0, 5, [6, 8, [7, 3, 1]], 9, 10]

print(find_item(list_input, 7))

此解决方案不需要 explicit 第三个参数。并在找不到项目时返回空的list，而不是None。注意使用新的 walrus 运算符：=

            if result := find_item(head, item):

如果有问题，请执行以下操作：

            result = find_item(head, item)
            if result:

在嵌套列表中查找索引

2 个答案: