Python - 在n维列表中查找所有项目x的位置

Question

如何找出所有项目 x 在任意形状或大小的 n 维列表中的位置？ （例如：[1, 2, [3, [1]]...]）

这是我找到第一项的代码：（未经过大量测试）

def where(x, inputList):
    return _where(x, inputList, [])
def _where(value, inputList, dimension):
    if isinstance(inputList, list):
        i = 0
        for l in inputList:
            dimension.append(i)
            D = _where(value, l, dimension)
            if not D == None:
                return D
            i += 1
            del dimension[-1]
        return None
    else:
        if value == inputList:
            return dimension
        else:
            return None

它递归地检查列表中的每个项目，当找到正确的项目时，它会返回该项目的维度或坐标

所需的输入/输出示例：

x = 1
inputlist = [1, [2, 21], [1]]
o = where_All(x, inputlist)
# o is [[0], [2, 0]]
print inputlist[0]    # is 1
print inputlist[2][0] # is 1

O 是列表中每个项目的坐标列表，等于 x

Answer 1

如果我理解正确，您想在嵌套数组中查找元素的坐标。然后，您可以使用以下更简单的函数并利用 yield，这在输入 (haystack) 本身是迭代器时特别有用（例如，从文件或类似文件中读取时）：

def where(needle, haystack, indexes=[]):
    for i, el in enumerate(haystack):
        if type(el) == list:
            for res in where(needle, el, indexes + [i]):
                yield res 
        elif el == needle:
            yield(indexes + [i])

a = [1, 2, [3, [1]]]
for coords in where(1, a):
    print(coords)

结果：

[0]
[2, 1, 0]