Python - Поиск местоположения всех элементов x в n-мерном списке

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 

вывода представляет собой список координат для каждый из них em в списке, который равен 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]