Python: итерация по каждому элементу в списке вложенных списков и замена определенных элементов
Я начинающий в python. Я хочу знать, есть ли встроенная функция или другой способ, поэтому я могу выполнить ниже в python 2.7:
Найдите все -letter в списке и подпишите и замените его ['not', letter]
Например: найдите все элементы в нижнем списке, начиная с - и замените их на ['not', letter]
Input : ['and', ['or', '-S', 'Q'], ['or', '-S', 'R'], ['or', ['or', '-Q', '-R'], '-S']]
Output : ['and', ['or', ['not','S'], 'Q'], ['or', ['not','S'], 'R'], ['or', ['or', ['not','Q'], ['not','R']], ['not','S']]]
Может кто-нибудь предложить, как это сделать в python.
Благодаря
Ответы
Ответ 1
Попробуйте немного рекурсии:
def change(lol):
for index,item in enumerate(lol):
if isinstance(item, list):
change(item)
elif item.startswith('-'):
lol[index] = ['not',item.split('-')[1]]
return lol
В действии:
In [24]: change(['and', ['or', '-S', 'Q'], ['or', '-S', 'R'], ['or', ['or', '-Q', '-R'], '-S']])
Out[24]:
['and',
['or', ['not', 'S'], 'Q'],
['or', ['not', 'S'], 'R'],
['or', ['or', ['not', 'Q'], ['not', 'R']], ['not', 'S']]]
Ответ 2
Вам нужно использовать рекурсивную функцию. isinstance(item, str)
просто проверяет, является ли элемент строкой.
def dumb_replace(lst):
for ind, item in enumerate(lst):
if isinstance(item, str):
if item.startswith('-'):
lst[ind] = ['not', 'letter']
else:
dumb_replace(item)
и
dumb_replace(Input)
дает:
['and', ['or', ['not', 'letter'], 'Q'], ['or', ['not', 'letter'], 'R'], ['or', ['or', ['not', 'letter'], ['not', 'letter']], ['not', 'letter']]]
Ответ 3
На основе найденного рецепта здесь:
def nested_list_replacer(seq, val = '-S', sub = ['not', 'letter']):
def top_kill(s):
for i in s:
if isinstance(i, str):
if i == val:
i = sub
yield i
else:
yield type(i)(top_kill(i))
return type(seq)(top_kill(seq))
l = ['and', ['or', '-S', 'Q'], ['or', '-S', 'R'], ['or', ['or', '-Q', '-R'], '-S']]
print(nested_list_replacer(l, '-S'))
print(nested_list_replacer(l, '-Q'))
дает:
['and', ['or', ['not', 'letter'], 'Q'], ['or', ['not', 'letter'], 'R'], ['or', ['or', '-Q', '-R'], ['not', 'letter']]]
['and', ['or', '-S', 'Q'], ['or', '-S', 'R'], ['or', ['or', ['not', 'letter'], '-R'], '-S']]