Понимание списков Python для создания нескольких списков

Question

Я хочу создать два списка listOfA и listOfB для хранения индексов A и B из другого списка.

s=['A','B','A','A','A','B','B'] 

Вывод должен быть два списка

listOfA=[0,2,3,4] 
listOfB=[1,5,6] 

Я могу сделать это с помощью двух операторов.

listOfA=[idx for idx,x in enumerate(s) if x=='A'] 
listOfB=[idx for idx,x in enumerate(s) if x=='B'] 

Однако, я хочу сделать это только в одной итерации, используя только списки. Можно ли сделать это в одном заявлении? что-то вроде listOfA,listOfB=[--code goes here--]

Answer 1

Само определение определения списка состоит в том, чтобы произвести один объект списка. Ваши 2 объекта списка имеют разную длину; вам придется использовать побочные эффекты для достижения того, чего вы хотите.

Не используйте здесь списки. Просто используйте обычный цикл:

listOfA, listOfB = [], [] 

for idx, x in enumerate(s): 
    target = listOfA if x == 'A' else listOfB 
    target.append(idx) 

Это оставляет вам только один цикла для выполнения; это побьет любые два понимания списков, по крайней мере, пока разработчики не найдут способ сделать переписку списков, строят список в два раза быстрее, чем цикл с отдельными вызовами list.append().

Я бы выбрал этот день в соответствии с вложенным списком только, чтобы иметь возможность создавать два списка на одной строке. Как указано в Zen of Python:

Показатели удобочитаемости.

Answer 2

Сортировка; ключ для генерации списка 2-элемент, который вы можете распаковывать:

listOfA, listOfB = [[idx for idx, x in enumerate(s) if x == c] for c in 'AB'] 

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

Answer 3

Приятным подходом к этой проблеме является использование defaultdict. Как уже сказал @Martin, понимание списка не является правильным инструментом для создания двух списков. Использование defaultdict позволит вам создать сегрегацию с использованием одной итерации. Кроме того, ваш код не будет ограничен в любой форме.

>>> from collections import defaultdict 
>>> s=['A','B','A','A','A','B','B'] 
>>> listOf = defaultdict(list) 
>>> for idx, elem in enumerate(s): 
    listOf[elem].append(idx) 
>>> listOf['A'], listOf['B'] 
([0, 2, 3, 4], [1, 5, 6])