Подсчитать глубину указанного c элемента во вложенном списке - PullRequest
0 голосов
/ 14 июля 2020

Я пытаюсь выяснить, как получить количество уровней определенного элемента во вложенном списке.

my_list = ["a", ["b", ["c", "d"], "e"], "f", ["g", ["h"]]]

Чтобы получить уровень элемента «e», я попытался создать функция для рекурсии, но не удалось ...

def get_level(letter, my_list):
    cnt = 0
    for sub_list in my_list:
        if letter in sub_list:
            cnt += 1
            return cnt
        else:
            get_level(letter, sub_list)

letter = "e"
print(get_level(letter, my_list))

Результат должен быть 2.

Пожалуйста, дайте мне знать, есть ли способ для этого.

1 Ответ

1 голос
/ 14 июля 2020

Получилось примерно так:

def find_e(arr, index):
  if 'e' in arr:
    return index
  else:
    for element in arr:
      if isinstance(element, list):
        return find_e(element, index + 1)

my_list = ["a", ["b", ["c", "d"], "e"], "f", ["g", ["h"]]]
print('Index is: ',find_e(my_list, 0))
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...