Разделение списка в Python

Question

Я пишу парсер на Python. Я преобразовал входную строку в список токенов, например:

['(', '2', '.', 'x', '.', '(', '3', '-', '1', ')', '+', '4', ')', '/', '3', '.', 'x', '^', '2']

Я хочу иметь возможность разбить список на несколько списков, например, функция str.split('+'). Но, похоже, нет способа сделать my_list.split('+'). Есть идеи?

Спасибо!

Answer 1

Вы можете написать свою собственную функцию разделения для списков довольно просто, используя yield:

def split_list(l, sep):
    current = []
    for x in l:
        if x == sep:
            yield current
            current = []
        else:
            current.append(x)
    yield current

Альтернативный способ - использовать list.index и перехватить исключение:

def split_list(l, sep):
    i = 0
    try:
        while True:
            j = l.index(sep, i)
            yield l[i:j]
            i = j + 1
    except ValueError:
        yield l[i:]

В любом случае вы можете назвать это так:

l = ['(', '2', '.', 'x', '.', '(', '3', '-', '1', ')', '+', '4', ')',
     '/', '3', '.', 'x', '^', '2']

for r in split_list(l, '+'):
    print r

Результат:

['(', '2', '.', 'x', '.', '(', '3', '-', '1', ')']
['4', ')', '/', '3', '.', 'x', '^', '2']

Для разбора в Python вы также можете посмотреть что-то вроде pyparsing .

Answer 2

быстрый взлом, вы можете сначала использовать метод .join (), чтобы присоединиться, создать строку из вашего списка, разделить ее на '+', повторно разделить (это создаст матрицу), затем использовать метод list () для дальнейшего разделения каждого элемента в матрице на отдельные токены

a = ['(', '2', '.', 'x', '.', '(', '3', '-', '1', ')', '+', '4', ')', '/', '3', '.', 'x', '^', '2']

b = ''.join(a).split('+')
c = []

for el in b:
    c.append(list(el))

print(c)

результат:

[['(', '2', '.', 'x', '.', '(', '3', '-', '1', ')'], ['4', ')', '/', '3', '.', 'x', '^', '2']]