Как организовать список в одну строку в Питоне?

Question

[Михаил Болеев]

Как правильно написать код вычисления списка в одну строку? Есть такой код:

a_list = [1, [2, 3]]
new_list = []

for el in a_list:
    if isinstance(el, list):
          for inner_el in el:
               new_list.append(inner_el)
    else:
          new_list.append(el)

Как правильно заменить его одной строкой? Следующие варианты не работают:

new_list = [inner_el if isinstance(el, list) else el for el in a_list for inner_el in el]

new_list = [el if not isinstance(el, list) else inner_el for el in a_list for inner_el in el]

Выдается ошибка: TypeError: 'int' object is not iterable

Comments

[MinTnt]

Что-то на подобии

print(sum(map(lambda x: x if isinstance(x, list) else [x], a_list), []))

Или через рекурсию?

[Михаил Болеев]

MinTnt, Спасибо. Почему в виде ответа не оформил?

[MinTnt]

Михаил Болеев ,
А вот в виде рекурсии. Хотя так оно даже несколько уровневый массив, до одномерного сведёт

a_list = [1, [2, [3]]]

def one_dimens_mass(lst):
	new_list = []
	for elem in lst:
		if type(elem)==list:
			new_list+= one_dimens_mass(elem)
		else:
			new_list.append(elem)
	return new_list
	
print(one_dimens_mass(a_list)

[Михаил Болеев]

MinTnt, Спасибо. Просто задача стояла использовать при распаковке заданное значение вложенности.

Answer 1

[Vindicar]

new_list = sum( (item if isinstance(item, list) else [item] for item in a_list) , [])

По сути, мы каждый элемент a_list превращаем в список, если он не был списком.
А потом суммируем все эти списки (это будет операция конкатенации).
Честно говоря, простой цикл for был бы читаемее (и быстрее, я думаю):

new_list = []
for item in a_list:
  if isinstance(item, list):
    new_list.extend(item)
  else:
    new_list.append(item)

Но будет работать только для одного уровня вложенности. Если нужно больше, используй рекурсию или что-то подобное.

Comments

[Михаил Болеев]

Спасибо. Для большей вложенности у меня работает внешний цикл while. Просто интересно стало, как можно подсократить код.

[MinTnt]

(item if isinstance(item, list) else [item] for item in a_list)

Как уже давно известно, и как однажды сам создатель языка сказал на какой-то конференции, генераторы работают медленнее чем map filter. Так что не рекомендую...

[o5a]

MinTnt, это не так. map будет быстрее, если использовать какую-то готовую функцию, например sum(map(int,values)). А в случае с lambda она будет как правило медленнее.

[MinTnt]

o5a, хм... Я просто решил вчера сделать замеры через функцию time(), и там всё оказалось быстрее

Answer 2

[MinTnt]

print(sum(map(lambda x: x if isinstance(x, list) else [x], a_list), []))

a_list = [1, [2, [3]]]

def one_dimens_mass(lst):
  new_list = []
  for elem in lst:
    if type(elem)==list:
      new_list+= one_dimens_mass(elem)
    else:
      new_list.append(elem)
  return new_list
  
print(one_dimens_mass(a_list)

Answer 3

[aRegius]

def flatten(items):
	  for i in items:
		    if isinstance(i, Iterable):
			      yield from flatten(i)
		    else:
			      yield i

Comments

[Михаил Болеев]

Это все интересно, конечно, но вопрос был в другом.

[aRegius]

Михаил Болеев, понимаю...