Как разложить вложенные массивы в одномерный?

Question

[Иван Четчасов]

Допустим у меня есть массив ['lloc', [['.', 'orgstart']]], и мне нужно рекурсивно разложить его в одномерное состояние, например: ['lloc', '.', 'orgstart'], также желательно сделать чистку точек: ['lloc', 'orgstart'], но это уже не обязательно.

Comments

[Vindicar]

В чём затруднение? Покажи свой код.

Answer 1

[Denis]

my_list = ['lloc', [['.', 'orgstart']]]


def my_function(array):
  res = ''
  for i in array:
    if type(i) != list and i != '.':
      res += i + ' '
    elif type(i) == list and i != '.':
      res += my_function(i)
  
  return res

result = my_function(my_list).split(' ')[:-1]
print(result) # -> ['lloc', 'orgstart']

Comments

[Иван Четчасов]

и без collections)))

[Пасечник Кузьмич]

Иван Четчасов, только очень медленно работает, жрёт память и не справится с другими типами последовательностей.

[Иван Четчасов]

Пасечник Кузьмич, мне другие типы и не нужны

[Иван Четчасов]

Пасечник Кузьмич, можно ещё и на cython переписать, если критически медленно работать будет

Answer 2

[Иван Четчасов]

Я сам решил задачу. В случае неверной структуры массива, возбуждается RuntimeError. Эта будет чуть оптимизированнее чем у denislysenko.

def flatten(array: list):
    def popen(arr: list | tuple):
        if not arr or not arr[0]:
            return arr
        while type(arr[0]) in (tuple, list):
            arr = arr[0]
            pass
        return arr
    if len(array) in (0, 1):
        return popen(array)
    if type(array[0]) == str:
        if type(array[1]) == list:
            return [array[0], *flatten(array[1:])]
            pass
        return array
    raise RuntimeError

Answer 3

[Сергей Горностаев]

from collections import abc

def flatten(iterable):
    for item in iterable:
        if isinstance(item, (str, bytes)):
            yield item
        elif isinstance(item, abc.Sequence):
            yield from flatten(item)
        else:
            yield item


data = ['lloc', [['.', 'orgstart']]]

flat_data = list(flatten(data))
clear_data = list(i for i in flatten(data) if i != '.')

Comments

[Иван Четчасов]

проблема: если передать в функцию, например, ["main"] как аргумент, то я получу ['m', 'a', 'i', 'n']. Как это исправить?

[Сергей Горностаев]

Иван Четчасов, с функцией всё в порядке, вы что-то не так делаете. Живой пример.

[Пасечник Кузьмич]

Сергей Горностаев, генератор фильтрующий точки не рассчитан на получение строки.

[Пасечник Кузьмич]

Хотя нет, не могу воспроизвести и с ним тоже. Автор вопроса наверняка что-то изменил в коде.

[Сергей Горностаев]

Пасечник Кузьмич, автор в функцию передаёт строку 'main' вместо списка ['main']. Ровно такой же результат он получит и с функцией из другого ответа.