Объединить Python список элементов на основе другого списка

Question

У меня есть 2 списка:

phon = ["A","R","K","H"] 
idx = [1,2,3,3] 

idx соответствует тому, как phon должны быть сгруппированы. В этом случае, phon_grouped должны быть ["A","R","KH"], потому что оба "K" и "H" соответствуют группе 3.

Я предполагаю, что какой-то zip или map функции требуется, но я не уверен, как это осуществить. У меня есть что-то вроде:

a = [] 
for i in enumerate(phon): 
    a[idx[i-1].append(phon[i]) 

, но это на самом деле не работает/компилировать

Answer 1

Используйте zip() и itertools.groupby() группе выход после проносясь:

from itertools import groupby 
from operator import itemgetter 

result = [''.join([c for i, c in group]) 
      for key, group in groupby(zip(idx, phon), itemgetter(0))] 

itertools.groupby() требует, чтобы ваш вход уже отсортирован на ключ (ваши idx значения здесь).

• zip() пар индексов вверх от idx с символами из phon
• itertools.groupby() групп в результате кортежей по первому значению, индекс. Значения равных индексов помещают кортежи в одну и ту же группу
• Последовательность списка затем выбирает символы из группы и объединяет их в строки.

Демо:

>>> from itertools import groupby 
>>> from operator import itemgetter 
>>> phon = ["A","R","K","H"] 
>>> idx = [1,2,3,3] 
>>> [''.join([c for i, c in group]) for key, group in groupby(zip(idx, phon), itemgetter(0))] 
['A', 'R', 'KH'] 

Answer 2

Если вы не хотите использовать дополнительный класс:

phon = ["A","R","K","H"] 
idx = [1,2,3,3] 
a = [[] for i in range(idx[-1])] # Create list of lists of length(max(idx)) 
for data,place in enumerate(idx): 
    a[place-1].append(phon[data]) 

[[ 'A'], [ 'R'], [ 'K ',' H ']]

В основном трюк состоит в том, чтобы просто предварительно инициализировать ваш список. Вы знаете, что окончательный список будет иметь максимальное число, найденное в idx, которое должно быть последним числом, как вы сказали, idx сортируется.

Не уверен, что если вы хотите, конечный результат будет литерой список, или сцепленные символы, т.е. «KH» против [ «K», «H»]